欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

faster-rcnn tensorflow CPU版代码与demo运行流程

程序员文章站 2024-03-14 08:48:04
...

faster-rcnn tensorflow CPU版代码与demo运行流程

最近尝试跑faster-rcnn代码,遇到好多坑,看了好多博客,曾一度想要放弃……(希望各位大佬不要笑我哇)我用CPU跑的别人的模型,github上有很多代码跑起来总会有这样那样的问题,最可怕是有的还必须用GPU,所以把我改过的代码和所有需要的东西都放在这,希望像我一样刚开始接触的同学运行demo时可以事半功倍。
这是我参考的代码,我是基于这份代码改的。

1、代码下载:

链接:https://pan.baidu.com/s/1KUK7pk4qG-OGc30anPd9AA 密码:loxv
数据集训练好的模型都在里面不需要再下载了。

2、运行环境:

  • deepin linux系统(使用windows的小伙伴别着急,装个linux系统其实挺简单的,而且操作跟windows差不多!)
  • python 3.7 (我用的这个,但是3.6/3.5也都行)+ tensorflow
  • 需要的包:cython opencv-python easydict pillow matplotlib scipy
    下图是我装的,有些可能没用到,因为我尝试运行过很多代码。
    faster-rcnn tensorflow CPU版代码与demo运行流程
    faster-rcnn tensorflow CPU版代码与demo运行流程
    faster-rcnn tensorflow CPU版代码与demo运行流程

3、运行demo

  • 进入lib编译(后面运行出错的话,解决后要重新编译)
cd lib
make clean
make
cd ..
  • 运行demo
python ./tools/demo.py
相关标签: object detection