py-faster-rcnn运行实例部分问题

来源:互联网 发布:淘宝上的xbox手柄 编辑:程序博客网 时间:2024/06/03 18:42

最近干cnn图像识别去了,很长时间没做ros.最近用py-faster-rcnn做了一个效果一般的识别例程,算是积累了一下经验。

samylee_py-faster-rcnn的demo.py测试

samylee_py-faster-rcnn自己制作数据集并且识别实体

samylee_自行训练一个数据集的例程

感谢samylee写的新手教程,基本按照s大的套路来东西就能做完。这里汇总一下自己遇到的问题。

(感谢L大写的脚本!不然感觉又要磨蹭好久……)

1.在测试vgg的时候参数问题 在命令行添加参数的时候一定要严格按照格式 在上面的第三个例程里,教程给出的参数是
./experiments/scripts/faster_rcnn_end2end.sh 0 VGG_CNN_M_1024 pascal_voc \ --set EXP_DIR foobar RNG_SEED 42 TRAIN.SCALES "[400, 500, 600, 700]"
其中后面脚本后的都是参数。”[400,500,600,700]”不应带空格

2.训练过程中报错object has no attribute 'text_format'
这个是google_protobuf版本冲突的问题。由于我之前安装了protobuf3.4,所以需要在./lib/fast_rcnn/train.py增加一行import google.protobuf.text_format 即可解决问题.感谢火腿肠工厂

3.后期训练出现object has no attribute 'text'
在github上找到的解答,感谢wuyuzaizai 在./lib/dataset/voc_eval.py里把没有标记的标签注释掉
obj_struct = {}
obj_struct['name'] = obj.find('name').text

# obj_struct['pose'] = obj.find('pose').text
# obj_struct['truncated'] = int(obj.find('truncated').text)
obj_struct['difficult'] = int(obj.find('difficult').text)
在自己准备的xml文件里,没有pose和truncated标签,所以就注释掉。

4.KeyError:max_overlaps
出现这个问题删除data文件夹里的cache即可

3.demo.py看训练结果
在./tools/demo.py里的demo函数,CONF_THRESH是匹配度阈值。调整这个数值就可以看读取的图片的识别结果。

暂时就这么多。

原创粉丝点击