#!/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.csdn.net/qq_27432889/article/details/52457011
caffe利用python接口绘制网络结构图
猜你喜欢
转载自blog.csdn.net/zlf19910726/article/details/80975132
今日推荐
周排行