faster Rcnn训练自己的数据集

数据集制作

http://blog.csdn.net/sinat_30071459/article/details/50723212

训练大体步骤 

http://blog.csdn.net/sinat_30071459/article/details/51332084

谢谢大神分享。

路过的坑

1.数据集为png图片

py-faster-rcnn/lib/datasets/pascal_voc.py

找到self._image_ext='.jpg' 换成'.png'

2.AttributeError: 'module' object has no attribute 'text_format'

需要再lib/fast_rcnn/train.py中添加：

import google.protobuf.text_format

3.gt_argmax_overlaps = overlaps.argmax(axis=0)

VauleError: attempt to get argmax of an empty sequence

数据集为行人，太过瘦长。没办法，我把照片宽拉长了两倍。测试的时候也拉长，最后在把检测结果缩回。。