ImportError: No module named caffe.proto解决办法

在用自己的数据训练基于caffe的SSD模型的时候,我们需要将图片数据转换成lmdb格式,用到的脚本文件是SSD源码里面提供的create_data.sh(具体位置在$CAFFE_ROOT/data/VOC0712/create_data.sh),将它copy一份,取一个自己的名字,比如creat_scenetext_data.sh,然后根据自己的实际情况修改里面的路径。

修改完,在执行命令 sh creat_data.sh的时候,有可能会出现以下error:

[lisiqi@l22-240-142 caffe]$ sh create_lmdbdata_scenetext.sh 
Traceback (most recent call last):
  File "/data/home/lisiqi/my_SSD/caffe/scripts/create_annoset.py", line 7, in <module>
    from caffe.proto import caffe_pb2
ImportError: No module named caffe.proto
Traceback (most recent call last):
  File "/data/home/lisiqi/my_SSD/caffe/scripts/create_annoset.py", line 7, in <module>
    from caffe.proto import caffe_pb2
ImportError: No module named caffe.proto

可能原因:这种情况一般是没有把caffe中的和python相关的内容的路径添加到python的编译路径中。

解决办法:执行命令 

export PYTHONPATH=$CAFFE_ROOT/python:$PYTHONPATH

其中,$CAFFE_ROOT是下载的SSD源码的路径,比如我的就是  /data/home/lisiqi/my_SSD/caffe

注意:上述方法只是一次性的,下次再次运行creat_scenetext_data.sh的时候,还要进行如上操作。

  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值