使用pydot 畫CNN網絡結構圖

使用pydot畫CNN結構圖的難點在於安裝pydotgit 要嚴格按照安裝順序進行安裝,百度搜索就能夠找到,成功安裝pydot後僅須要兩行代碼就能夠畫出網絡結構github 前提是要搭建好網絡結構網絡 import pydotorm plot_model(model, to_file='model.png', show_shapes=True, show_layer_names=False)get
相關文章
相關標籤/搜索