caffe 绘制网络模型

Posted xiaoheizi

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了caffe 绘制网络模型相关的知识,希望对你有一定的参考价值。

参考网址:blog.csdn.net/u013989576/article/details/61618454

画网络模型有两个选择:

1.使用caffe提供的draw_net.py

2.使用Netscope可视化

使用第一种方法在服务器上做的时候,因为protoc的版本原因,没有好,现在突然想到可以在本地试一试,换个系统后再试一下。

如果要绘制caffe里的Lenet模型,可使用一下语句:

sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT
其中的netImage文件夹是要自己在相应目录新建的

使用netscope时,使用网址http://ethereon.github.io/netscope/#/editor,具体使用方法网址上也很清晰。但是会遇到部分模块‘翻译’不出来的现象,不知是不是浏览器版本的原因,用了谷歌和搜狗效果都是一样

 

可视化后才能更清晰的理解整个过程。

以上是关于caffe 绘制网络模型的主要内容,如果未能解决你的问题,请参考以下文章

Caffe---自带工具 绘制loss和accuracy曲线

如何将经过训练的 caffe 模型以 h5 格式加载到 c++ caffe 网络?

Caffe学习系列——工具篇:神经网络模型结构可视化

caffe和vgg-16和googlenet的区别

使用caffe的cifar10网络模型训练自己的图片数据

一文带你熟悉Pytorch-;Caffe-;om模型转换流程