OptimizedMakeNN

• 簡介
  • 原始項目地址 https://github.com/rtygbwwwerr/MakeNN1.git。
  • 這是一個輕量的繪制神經網絡結構圖（主要神經元結構）的繪圖工具包，基于graphviz。
  • 在這個項目中，我優化了一些代碼（由Python2遷移Python3） ，使之在命令行下使用更加方便。
• 安裝
  • 使用git clone git@github.com:luanshiyinyang/OptimizedMakeNN.git
  • 請一定安裝graphviz軟件并配置環境變量
• 說明
  • 原始項目由rtygbwwwerr完成,在原始項目中你可以使用下面的命令行命令繪制神經元結構圖。
    • python src/makeNN.py config.txt
  • config.txt 是生成圖結構的配置文件, 使用utf-8編碼修改即可生成自定義圖片。
    • 這里其實是對graphviz dot語言神經元繪圖的模式做了封裝，如果很擅長dot語言，可以直接使用其編寫腳本。
    • 這個文件中每一行表示一層，具體格式必須按照config.txt文件示例。（編碼一定要utf-8，標點使用英文標點）
      • 層名稱
      • 神經元個數
      • 神經元上色顏色
      • 該層與下一層的神經元連接
        • 節點noden_m（如第1層輸入層第二個節點到第一個隱層第二個節點node0_1->node1_1）
        • all->all表示全連接
        • None->None表示無連接，一般為最后一層
  • 命令行參數
    • -i --input 輸入的gv文件（dot語言編寫，不輸入則依據config.txt生成）
    • -c --config 輸入的生成gv文件的腳本
    • -o --output 輸出png圖片目錄（如output.rst.png、rst.png等）
    • 注意
      • 上述-i與-c沖突，不要同時輸入（否則優先按照-i的gv文件生成圖片）
• 使用示例
  • python main.py --output rst.png
  • 配置文件如下
  • 結果如下
• 補充說明
  • 本項目適合神經元繪圖，深層神經網絡（如卷積神經網絡），這個工具并不適合你。
  • 目前只在windows環境下測試完成。
  • 本項目開源于我的Github，歡迎star或者fork。

總結

以上是生活随笔為你收集整理的深度学习项目-神经元结构可视化的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。