caffe的网络结构存储在prototxt中,如何可视化网络?

来源:互联网 发布:战意首测激活码淘宝 编辑:程序博客网 时间:2024/06/08 04:49

可视化网络需要一个py代码:draw_net.py

进入存放draw_net.py的目录下, $  ./draw_net.py --rankdir TB ./lenet_train_test.prototxt mnist.png       

#TB代表从top画到bottom,lenet_train_test.prototxt是lenet的网络结构,存储成名字为mnist.png的图片


0 0