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

猜你喜欢

转载自blog.csdn.net/u014285910/article/details/86521518