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

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

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
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值