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
下图是我装的,有些可能没用到,因为我尝试运行过很多代码。
3、运行demo
- 进入lib编译(后面运行出错的话,解决后要重新编译)
cd lib
make clean
make
cd ..
- 运行demo
python ./tools/demo.py