您的位置:首页 > 理论基础 > 计算机网络

解决Caffe不能绘制网络模型的错误

2017-09-29 14:57 162 查看
最近在尝试使用caffe自带的draw_net.py绘制网络模型图,但是遇到了错误:

AttributeError: 'google.protobuf.pyext._message.RepeatedScalarConta' object has no attribute '_values'

找了好多方法,都说是peotobuf版本不对,重装了protobuf还是不行,最后改了下Caffe_root/python/caffe/draw.py中的代码,问题解决。

建议:

1.先查看一下自己的protobuf版本 protoc --version,看是否满足caffe_root/python/requirements.txt中的要求,有小于3.0(caffe之前版本不支持3.0以上的protobuf,最新的版本不清楚),如果版本太高或者太低,就重装一个protobuf

2.如果protobuf版本正确,还是存在该错误,那就进入Caffe_root/python/caffe/draw.py中,找到函数get_layer_label中,把对应的几个._value删掉,保存,再运行draw_net.py就好了!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息