caffe绘制神经网络图
caffe绘制神经网络图
0. 前言
平时都是用pytorch的,但领了用caffe画图的任务,遂玩一下
1. 尝试anaconda安装coffe中遇到的问题
提前说明最后也没解决,但总结了过程中查到的方法
在anaconda环境下执行命令:
conda install caffe
报错PackagesNotFoundError,查询资料,尝试了,均无效
换源
恢复默认源
conda config --remove-key channels
清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --set show_channel_urls yes
更新conda
conda update -n base conda
conda update --all
换指令
conda config --set channel_priority flexible
conda install -c anaconda3/ caffe
换python版本
在anaconda中新建python2.7版本
anaconda search安装
anaconda search -t conda caffe
搜出来后找win64版本,然后输入
anaconda show 名称
复制给的指令,执行
最后放弃了,用linux装
题外话:Netscope在线可视化caffe网络
http://ethereon.github.io/netscope/quickstart.html
在这个网址里把caffe网络的prototxt文件拷贝进去shift+enter即可生成网络图
可以用这个网络测试:
https://github.com/yihui-he/resnet-imagenet-caffe/blob/master/resnet_50/ResNet-50-test.prototxt
效果:
展示的还蛮清晰的,感觉作为系统的外部链接挺好
3. linux下caffe绘图
安装相关包
conda install caffe
conda install GraphViz
conda install pydot
python /root/anaconda3/bin/draw_net ResNet-50-test.prototxt(prototxt的路径) 666.png(保存图片的路径)
这里值得注意的是命令里一般用的是draw_net.py,但使用anacodna的话需要如上命令直接执行anaconda里的文件。
然而,还是报错:
google.protobuf.text_format.ParseError: 159:8 : Expected "}".
错误原因可能是protoc版本问题
whereis protoc # 查询protobuf安装位置
which protoc # 查询protobuf版本
然后尝试了各种更新版本,无果
最后发现protoc根本没问题,换了个简单点的vgg网络就跑出来了
应该是我之前没用过caffe少了什么东西吧
果然是程序员的日常啊…被不熟悉的框架玩的好惨