在containter的bash中,我們使用:
python caffe/python/draw _ net . py workspace/pretty net . proto txt workspace/pretty net . jpg 11
workspace/pretty net . proto txt:proto txt文件。
Workspace/prettynet.jpg:目標ipg的名稱(和位置)。
或者代碼。
#加載模塊
進口咖啡
進口咖啡. draw
從caffe.proto導入caffe_pb2
從google.protobuf導入文本格式
#設置參數
輸入網絡協議文件= '測試網絡協議文件' #網絡配置文件
output _ image _ file = ' test _ net . jpg ' #輸出圖像
rankdir ='LR' # net排列方法:LR、TB、RL
#閱讀網絡
net = caffe_pb2。網絡參數()
文本_格式。merge(open(input _ net _ proto _ file)。read()、net)
#收網
打印('繪制網絡到%s' % output_image_file ')
caffe . draw . draw _ net _ to _ file(net,output_image_file,rankdir)
打印('完成...')