ImportError: No module named caffe.proto解決辦法


 

原文   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:

 

[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的時候,還要進行如上操作。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM