1. 程式人生 > >繪製Caffe prototxt中網路結構圖

繪製Caffe prototxt中網路結構圖

1. 安裝相關依賴

  • 安裝caffe
  • 安裝graphviz
sudo apt-get install graphviz  # 未安裝可能出現 OSError: [Errno 2] "dot" not found in path.
  • 安裝pydot
sudo pip install pydot

2. 繪製圖

  • 進入到prototxt所在目錄下
  • 繪圖
python /python/to/caffe/python/draw_net.py net.prototxt img.png --rankdir==BT