使用py-faster-rcnn训练自己的数据集

来源:互联网 发布:淘宝客佣金计算规范 编辑:程序博客网 时间:2024/05/16 12:49

具体流程按照这篇博客的内容来的:
http://blog.csdn.net/sinat_30071459/article/details/51332084
但是在训练过程中还是出现了一些问题,建议提前将这些改好再训练。
1. 首先是这个问题:
b2.text_format.Merge(f.read(), self.solver_param) AttributeError: ‘module’ object has no attribute ‘text_format’
解决方法:在文件./lib/fast_rcnn/train.py增加一行import google.protobuf.text_format 即可解决问题

2.然后又出现了这个问题
TypeError numpy.float64 object cannot be interpreted as an index
解决方法:有文章说把numpy的版本由高版本如1.12.0降为1.11.0问题得以解决。
sudo pip install -U numpy==1.11.0
但是用了这种方法还是有问题,我又将其升级到高版本,按照这篇文章的方法将语法不对的地方做了修改
http://blog.csdn.net/a8039974/article/details/77628717
除了这篇文章提到的地方,还需要修改下面这些错误提示的地方,修改方法和上面相同
File “/home/zhouqin/py-faster-rcnn/tools/../lib/roi_data_layer/minibatch.py”, line 125, in _sample_rois
roidb[‘bbox_targets’][keep_inds, :], num_classes)
File “/home/zhouqin/py-faster-rcnn/tools/../lib/roi_data_layer/minibatch.py”, line 176, in _get_bbox_regression_labels
bbox_targets[ind, start:end] = bbox_target_data[ind, 1:]
TypeError: slice indices must be integers or None or have an index method

3.还有就是这个问题
F1221 12:36:42.488387 16329 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0) out of memory
大概是电脑配置不够,按照这篇文章的方法来修改就可以运行了
http://blog.csdn.net/u012283902/article/details/63255301

注:还有一些值得参考的博文
http://blog.csdn.net/awq520tt1314/article/details/72629867

原创粉丝点击