Pycaffe---進(jìn)行網(wǎng)絡(luò)結(jié)構(gòu)(xxx.prototxt)可視化

解決網(wǎng)絡(luò)結(jié)構(gòu)(xxx.prototxt)可視化，還可以借助python接口，編寫一個(gè)類似如下的pycaffe_draw_net.py:

# -*- coding: utf-8 -*- #!/usr/bin/env python import caffe import caffe.draw from caffe.proto import caffe_pb2 from google.protobuf import text_format # Set the parameters input_net_file = '/home/wp/caffe/myself/road/prototxt_files/train_val.prototxt' # net configuration file output_image_file ='/home/wp/caffe/myself/road/Log/road_net.jpg' # the output image rankdir ='LR' # net arrange method: LR、TB、RL # Read the net net = caffe_pb2.NetParameter() text_format.Merge(open(input_net_file).read(), net) # Draw the net print('Drawing net to %s' % output_image_file) caffe.draw.draw_net_to_file(net, output_image_file, rankdir) print('done...')

?

# cd caffe

# python? ./xx/xx/pycaffe_draw_net.py

?

轉(zhuǎn)載于:https://www.cnblogs.com/carle-09/p/9087101.html

總結(jié)

以上是生活随笔為你收集整理的Caffe---Pycaffe进行网络结构(xxx.prototxt)可视化的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。