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

Posted 2021-01-31 sddai

 

原文   https://blog.csdn.net/lanyuelvyun/article/details/73628152

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

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

 

[[email protected] 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的时候，还要进行如上操作。