caffe利用python接口绘制网络结构图

最新推荐文章于 2024-07-28 21:59:03 发布

CodingAlgo

最新推荐文章于 2024-07-28 21:59:03 发布

阅读量189

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习(基于Caffe)

本文链接：https://blog.youkuaiyun.com/zlf19910726/article/details/80975132

深度学习(基于Caffe) 专栏收录该内容

16 篇文章

订阅专栏

本文介绍如何使用Python结合PyCaffe库读取并绘制深度学习网络结构图，通过具体的代码实例展示了从加载网络配置文件到生成可视化图像的全过程。

#!/usr/bin/env python
#coding=utf-8

import sys
caffe_root = "D:\\Caffe\\caffe-windows\\Build\\x64\\Debug\\pycaffe"
sys.path.insert(0, caffe_root)

import caffe
import caffe.draw
from caffe.proto import caffe_pb2

from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
from google.protobuf import text_format

print('finish import!')


# Set parameters
net_file = 'D:/1byav/MTCNN/test_only/codes/camera_demo/model/det1.prototxt'  
image_file ='D:/1byav/MTCNN/test_only/codes/camera_demo/model/cifa10.png'  
rankdir ='LR'  

# Read the net
net = caffe_pb2.NetParameter()
text_format.Merge(open(net_file).read(), net)

print(net)
caffe.draw.draw_net_to_file(net, image_file, rankdir)
print('draw finish!')

参考： https://blog.youkuaiyun.com/qq_27432889/article/details/52457011