您的位置:首页 > 其它

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

2017-12-21 22:00 621 查看
具体流程按照这篇博客的内容来的:

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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  数据 博客