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少了什么东西吧
果然是程序员的日常啊…被不熟悉的框架玩的好惨