https://github.com/smallcorgi/Faster-RCNN_TF
1、首先放上源码,按照源码中执行第1、2两步。运行demo的时候执行如下代码,否则会出错
python ./tools/demo.py --model demo_model/VGGnet_fast_rcnn_iter_70000.ckpt
这里引用了https://www.jianshu.com/p/bb2933863a55作者到博客,非常感谢
2、训练模型,下载好VOC数据集后运行
./experiments/scripts/faster_rcnn_end2end.sh gpu 3 VGG16 pascal_voc
这里gpu 3可以换成你的gpu 代号(比如我的是1)运行后出现如下内容,证明运行成功。
3.之后运行:python ./tools/test_net.py --device gpu --device_id 0 --weights output/faster_rcnn_end2end/voc_2007_trainval/VGGnet_fast_rcnn_iter_70000.ckpt --imdb voc_2007_test --cfg experiments/cfgs/faster_rcnn_end2end.yml --network VGGnet_test 会报错
一直重复出现waiting for output.........
运行成功,接下来考虑运行自己的数据集啦