caffe模型的可視化

命令:

python path\caffe-master\python\draw_net.py --rankdir TB path\caffe-master\examples\mnist\lenet_train_test.prototxt path\caffe-master\examples\mnist\draw_net\mnist_tb.png
Drawing net to path\caffe-master\examples\mnist\draw_net\mnist_tb.png

總體來說就是:

python draw_net.py路徑 --rankdir TB 模型文件路徑 生成圖片路徑

最終的生成結果:
在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章