環(huán)境:win10+caffe2+python(VSCode)
下面這個方法是網(wǎng)上常見的方法,但是不知道為什么功茴,這種方法在我的環(huán)境下一直無法正常顯示圖片陨簇,后來索性直接將網(wǎng)絡模型保存至本地妒牙,用Opencv來顯示模型彼哼,網(wǎng)上常見的做法:
1.首先需要安裝graphviz,并配置系統(tǒng)環(huán)境變量
2.pip install pygraphviz
3.pip install pydot-ng
利用opencv來顯示網(wǎng)絡模型的方法:
1.創(chuàng)建網(wǎng)絡:
from caffe2.python import core,workspacefrom caffe2.proto
import caffe2_pb2
from caffe2.python import net_drawer
import cv2
X = net.GaussianFill([],["X"],mean=0.0,std=1.0,shape=[2,3],run_once=0)
print("new Network Porto:\n{}".format(net.Proto()))
W = net.GaussianFill([],["W"],std=1.0,mean = 0.0,shape=[5,3],run_once=0)
b = net.ConstantFill([],["b"],shape = [5,],value = 1.0,run_once=0)
Y = net.FC([X,W,b],["Y"])
print("current Network proto :\n{}".format(net.Proto()))
2.利用net_drawer的模塊將net轉化為一個pydot.Dot的類(本人不了解pydot模塊,有錯誤還望指出)湘今,該類可以有方法生成png文件:
graph = net_drawer.GetPydotGraph(net,rankdir="LR") #此處返回的應是pydot.Dot類
graph.write_png("out.png")#使用該方法可以生成圖片
接著就可以用opencv來顯示圖片了:
img1 = cv2.imread("out.png",1)
cv2.imshow("Netgraph",img1)
cv2.waitKey(0)