在ubuntu14.04下用自己的数据集训练faster-rcnn遇到的问题

来源:互联网 发布:淘宝店铺装修怎么还原 编辑:程序博客网 时间:2024/05/29 15:40

在下面这篇文章的指导下进行训练

http://blog.csdn.net/princepaul3/article/details/52313582#reply

(1)

在py-faster-rcnn根目录下执行:

./experiments/scripts/faster_rcnn_alt_opt.sh 0 ZF pascal_voc

出现下面的错误:

 

pocess Process-1:
Traceback (most recent call last):
File "/usr/lib/python2.7/multiprocessing/process.py", line 258, in _bootstrap
self.run()
File "/usr/lib/python2.7/multiprocessing/process.py", line 114, in run
self._target(*self._args, **self._kwargs)
File "./tools/train_faster_rcnn_alt_opt.py", line 129, in train_rpn
max_iters=max_iters)
File "/home/caijiao/caffe/py-faster-rcnn/tools/../lib/fast_rcnn/train.py", line 158, in train_net
pretrained_model=pretrained_model)
File "/home/caijiao/caffe/py-faster-rcnn/tools/../lib/fast_rcnn/train.py", line 52, in __init__
pb2.text_format.Merge(f.read(), self.solver_param)
AttributeError: 'module' object has no attribute 'text_format'

解决方法是:

 改一下train.py源码文件,把import google.probuf as pb2 改成 import google.probuf.text_format as pb2 
在train.py后面代码中,将pb2.text_format改成pb2 
 
(2)由于我的图像是从Imagenet下载下来的,图片的后缀是.JPEG,在做图像标注的时候也没注意到这个问题,后面在做训练的时候,发现出错,
需要在以下进行修改:
在pascal_voc里改
self._unage_ext='.jpg'改成self._unage_ext='.JPEG'
 
(3)训练了一个多小时后出现如下错误:

解决办法是:numpy不要用1.12,需要用1.11

电脑里有2个版本,可能无法识别,卸载了1.12,问题解决。

0 0
原创粉丝点击