DIGITS運行時候遇到PicklingError的解決方法

在加載DIGITS時,由於Caffe的更新對命名空間caffe_pb2的修改會導致以下錯誤:

Can't pickle <class 'caffe.proto.caffe_pb2.NetParameter'>: it's not the same object as caffe.proto.caffe_pb2.NetParameter

修改的方法是: caffe_pb2模塊應作爲子模塊導入,而不是原來的獨立模塊導入,具體的修改方法參考一下的網頁:

https://github.com/uhfband/DIGITS/commit/59faee35442191356e43f51df87bd9253c1f6f1f#diff-0d1ba4e35115631664436b13944953fd

如果出現digits在訓練的時候,網頁中沒有反應圖形化顯示,但是在控制檯中的顯示正常的情況,則需要在啓動服務起之前,在命令控制行中,導入路徑:

export PYTHONPATH=${CAFFE_HOME}/python/caffe/proto:$PYTHONPATH

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章