mx.viz.plot_network，無法繪製RESNET

傢伙，我從這個網址下載RESNET-152：http://data.mxnet.io/models/imagenet-11k/resnet-152/mx.viz.plot_network，無法繪製RESNET

然後，我嘗試繪圖本RESNET有152層以這樣的方式

import mxnet as mx 

sym, arg_params, aux_params = mx.model.load_checkpoint('resnet-152', 0) 
pltn = mx.viz.plot_network(sym, shape = {'data': (1, 3, 224, 224)}, \ 
node_attrs = {'shape': 'oval', 'fixedsize': 'false'}) 
pltn.view('resnet-152')

但是，輸出pdf文件是空的。我可以繪製LeNet，AlexNet和VGG，但是resnet有什麼問題？非常感謝你。

來源

2017-01-10 tsuifrank

你的代碼適合我。生成一個150k PDF，所以它不是很大。你確定PDF實際上是空白的嗎？圖像的左上部分（可能是默認視圖）全白，因爲巨型網絡從頁面的左下角到右上角渲染。 – Leopd

非常感謝。它與「save_format」有一些關係。如果我設置save_format ='eps'，我可以看到情節。但是，情節只顯示了網絡的一部分，也許是因爲網絡太大。你能看到整個神經網絡嗎？謝謝。 – tsuifrank

我可以確認編寫的代碼與MXNet 1.1.0一起工作，並輸出包含整個神經網絡圖的PDF。

來源

2018-03-03 02:09:56

mx.viz.plot_network，無法繪製RESNET

回答

相關問題