Caffe 的視覺化 (二)網路結構視覺化

2021-08-11 03:47:45 字數 728 閱讀 4282

caffe 的視覺化 (二)網路結構視覺化

使用caffe裡的工具 draw_net.py   (該檔案在目錄 $(caffe_root)/python/ 裡)

usage: draw_net.py [-h] [--rankdir rankdir] [--phase phase]

input_net_proto_file  output_image_file

以lenet為例:

mark@ubuntu:~/caffe$ python  ./python/draw_net.py    ./examples/mnist/lenet_train_test.prototxt    mnist.png

生成的網路結構圖 mnist.png 如下:

也可以加上引數 rankdir

mark@ubuntu:~/caffe$ python   ./python/draw_net.py   --rankdir   bt   ./examples/mnist/lenet_train_test.prototxt   mnist2.png

生成的網路結構圖 mnist2.png 如下:

還可以使用下面的**來檢視網路結構圖

caffe中網路結構的視覺化

實驗工具是 ubuntu系統下的caffe 視覺化方法很多,一種是用caffe自帶的draw net.py 來實現網路結構的視覺化。具體實現如下 python caffe python draw net.py train.prototxt net.png用python命令執行draw net.py ...

Caffe小玩意 1 視覺化網路結構

最近在學習caffe,但是作為曾經的windows深度使用者,還是比較習慣視覺化的介面。然而,caffe當然是在linux os x系統下更好啦,因為一般還是寫script在命令列裡面玩的。所以這樣就不直觀咯,為了能直觀地看清楚網路結構,而不是看prototxt腦補 視覺化就很重要了。幸好,開發ca...

Pytorch網路結構視覺化

現在用的這個 net load model net,args.trained model,args.cpu for name,param in net.named parameters print param.size name,下面這個報錯 pytorch網路結構視覺化 使用這個 超級強悍 可以通...