caffe ssd 錯誤描述:
AttributeError: 'module' object has no attribute 'LabelMap'
SSD from caffe.proto import caffe_pb2 ImportError: No module named caffe.proto 解決方法
在用voc2007和voc2012的數據訓練基於caffe的SSD模型的時候,我們需要將圖片數據轉換成lmdb格式,用到的腳本文件是SSD源碼里面提供的create_data.sh(具體位置在$CAFFE_ROOT/data/VOC0712/create_data.sh)。
修改完,在執行命令 sh create_data.sh的時候,有可能會出現以下error:
因為這個問題在晚上弄了一個下午,嘗試過例如:https://blog.csdn.net/lanyuelvyun/article/details/73628152 等博客所說的方法,就是把把caffe中的和python相關的內容的路徑添加到python的編譯路徑中。然而這樣的方法並沒有用,所以糾結了一個下午。
解決方法:
1.打開 ~/caffe-ssd/scripts/create_annoset.py 文件。
2.在import sys和from caffe.proto import caffe_pb2這兩段代碼中間插入sys.path.insert(0,’/home/xxx/caffe_ssd/python’)這句代碼,xxx為你們自己的用戶名,/home/xxx/caffe_ssd就是caffe-ssd的根目錄。
然后保存文件。在執行命令 sudo ./data/VOC0712/create_data.sh 的時候就順利通過了。
---------------------
作者:curious999
來源:CSDN
原文:https://blog.csdn.net/curious999/article/details/81225624
版權聲明:本文為博主原創文章,轉載請附上博文鏈接!