1.GitHub - ajtulloch/dnngraph: A DSL for deep neural networks, supporting Caffe and Torch 2.caffe/draw_net.py at master · BVLC/caffe · GitHub 3.GitHub - gwding/draw_convnet . @蓋蚊叮,作者直接回答了這個問題,我搬運過來了。
MXNet(GitHub - dmlc/mxnet: Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Go, Javascript and more )這個優秀的深度學習工具自帶了一個可視化組件,參考github代碼 mxnet/visualization.py at master · dmlc/mxnet · GitHub 或者這片blog Visualizing CNN architectures side by side with mxnet 簡單的幾行代碼輕鬆畫出inception-v3(出處在左邊的那個blog,請自行點進去查看,圖太長貼知乎會影響排版被舉報)
import find_mxnet
import mxnet as mx
import importlib
name = "inception-v3"
net = importlib.import_module("symbol_" + name).get_symbol(2)
a = mx.viz.plot_network(net, shape={"data":(1, 1, 299, 299)}, node_attrs={"shape":"rect","fixedsize":"false"})
a.render(name)