Caffe学习笔记(2)--spyder 下绘制网络结构

最新推荐文章于 2020-04-02 21:32:38 发布

spcq4

最新推荐文章于 2020-04-02 21:32:38 发布

阅读量955

点赞数

分类专栏： caffe学习笔记文章标签： python caffe spyder 网络结构

本文链接：https://blog.csdn.net/spcq4/article/details/52678531

版权

本文介绍了如何在Spyder环境下使用Python脚本绘制Caffe的网络结构图，提供了一种在Anaconda集成开发环境中操作Caffe的方法。通过运行程序，网络结构图将保存至指定路径。

摘要由CSDN通过智能技术生成

直接使用Caffe中的python脚本绘制网络结构的方法请参照链接：http://www.cnblogs.com/denny402/p/5106764.html。因为本人在学习caffe 的时候希望在anaconda的环境下区编辑，所以这里介绍如何在spyder中编写python 程序来绘制网络结构图。程序如下：

#将caffe包含到路径中
import sys
caffe_home = '/home/kelly/DL/caffe-master/'
sys.path.insert(0, caffe_home + 'python')
#引用库
import caffe
import caffe.draw
from caffe.proto import caffe_pb2
from google.protobuf import text_format