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