項(xiàng)目地址

目前有很多工具可以繪制卷積神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)圖，我在使用過(guò)程中發(fā)現(xiàn)PlotNeuralNet在配置好環(huán)境后使用起來(lái)較為方便，且繪制結(jié)果比較美觀，該項(xiàng)目的地址為：PlotNeuralNet源代碼。
不過(guò)我在Windows系統(tǒng)下實(shí)際使用過(guò)程中發(fā)現(xiàn)一些問(wèn)題，因此對(duì)源代碼進(jìn)行了一些修改，修改后的項(xiàng)目地址為：lazyn的PlotNeuralNet，CSDN資源地址為卷積神經(jīng)網(wǎng)絡(luò)+網(wǎng)絡(luò)結(jié)構(gòu)+繪制網(wǎng)絡(luò)結(jié)構(gòu)圖。

官方教程

用戶克隆項(xiàng)目到本地后，根據(jù)官方教程，只需要下面兩步便可以繪制神經(jīng)網(wǎng)絡(luò)的結(jié)構(gòu)圖。

Install the following packages.

• Ubuntu 16.04

  sudo apt-get install texlive-latex-extra

• Ubuntu 18.04.2
  Base on this website, please install the following packages.

  sudo apt-get install texlive-latex-base sudo apt-get install texlive-fonts-recommended sudo apt-get install texlive-fonts-extra sudo apt-get install texlive-latex-extra

• Windows

Download and install MikTeX.

Download and install bash runner on Windows, recommends Git bash or Cygwin(https://www.cygwin.com/)

Execute the example as followed.

cd pyexamples/ bash ../tikzmake.sh test_simple

源代碼修改

首先在Windows環(huán)境下運(yùn)行腳本時(shí)，會(huì)出現(xiàn)../tikzmake.sh: line 13: xdg-open: command not found這樣的錯(cuò)誤，因此對(duì)腳本進(jìn)行如下修改。

其次源代碼僅支持filter尺寸為正方形的網(wǎng)絡(luò)顯示，盡管通過(guò) height=48, depth=48可以修改寬和高，但修改的也僅僅是顯示效果，實(shí)際特征圖尺寸仍為正方形，這樣輸出的網(wǎng)絡(luò)圖如下圖所示，該網(wǎng)絡(luò)圖還存在如下問(wèn)題：

• 特征圖尺寸顯示在邊角，在我看來(lái)不夠美觀；
• 無(wú)法顯示池化層特征圖信息；

為了解決上述問(wèn)題，對(duì)layers/Box.sty和pycore/tikzeng.py進(jìn)行修改，對(duì)比圖如下所示：

Box.sty

tikzeng.py

經(jīng)過(guò)上述修改后，輸出網(wǎng)絡(luò)圖如下所示，由于使用需要，我只修改了卷積層和池化層的代碼

總結(jié)

以上是生活随笔為你收集整理的PlotNeuralNet绘制卷积神经网络结构图的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。