caffe網絡結構圖繪製

繪製網絡圖通常有兩種方法: 一種是利用python自帶的draw_net.py,首先安裝兩個庫: sudo apt-get install graphviz sudo pip install pydot 接下來就可以用python自帶的draw_net.py文件來繪製網絡圖了。 draw_net.py執行時帶三個參數: 一、網絡模型的prototxt文件; 二、保存的圖片路徑和名字; 三、--ra
相關文章
相關標籤/搜索