酷炫的深度學(xué)習(xí)網(wǎng)絡(luò)圖怎么繪制出來的?
點(diǎn)擊上方“小白學(xué)視覺”,選擇加"星標(biāo)"或“置頂”
重磅干貨,第一時間送達(dá)
本文我們聊聊如何才能畫出炫酷高大上的神經(jīng)網(wǎng)絡(luò)圖,下面是常用的幾種工具。
作者&編輯 | 言有三
這個工具可以非常方便的畫出各種類型的圖,是下面這位小哥哥開發(fā)的,來自于麻省理工學(xué)院弗蘭克爾生物工程實(shí)驗(yàn)室, 該實(shí)驗(yàn)室開發(fā)可視化和機(jī)器學(xué)習(xí)工具用于分析生物數(shù)據(jù)。

github地址:https://github.com/zfrenchee
畫圖工具體驗(yàn)地址:http://alexlenail.me/NN-SVG/
可以繪制的圖包括以節(jié)點(diǎn)形式展示的FCNN style,這個特別適合傳統(tǒng)的全連接神經(jīng)網(wǎng)絡(luò)的繪制。

以平鋪網(wǎng)絡(luò)結(jié)構(gòu)展示的LeNet style,用二維的方式,適合查看每一層featuremap的大小和通道數(shù)目。

以三維block形式展現(xiàn)的AlexNet style,可以更加真實(shí)地展示卷積過程中高維數(shù)據(jù)的尺度的變化,目前只支持卷積層和全連接層。

這個工具可以導(dǎo)出非常高清的SVG圖,值得體驗(yàn)。
這個工具是薩爾大學(xué)計算機(jī)科學(xué)專業(yè)的一個學(xué)生開發(fā)的,一看就像計算機(jī)學(xué)院的嘛。

首先我們看看效果,其github鏈接如下,將近4000 star:
https://github.com/HarisIqbal88/PlotNeuralNet
看看人家這個fcn-8的可視化圖,顏值奇高。

使用的門檻相對來說就高一些了,用LaTex語言編輯,所以可以發(fā)揮的空間就大了,你看下面這個softmax層,這就是會寫代碼的優(yōu)勢了。

其中的一部分代碼是這樣的,會寫嗎。
\pic[shift={(0,0,0)}] at (0,0,0) {Box={name=crp1,caption=SoftmaxLoss: $E_\mathcal{S}$ ,%
fill={rgb:blue,1.5;red,3.5;green,3.5;white,5},opacity=0.5,height=20,width=7,depth=20}};
相似的工具還有:https://github.com/jettan/tikz_cnn
ConvNetDraw是一個使用配置命令的CNN神經(jīng)網(wǎng)絡(luò)畫圖工具,開發(fā)者是香港的一位程序員,Cédric cbovar。

采用如下的語法直接配置網(wǎng)絡(luò),可以簡單調(diào)整x,y,z等3個維度,github鏈接如下:
https://cbovar.github.io/ConvNetDraw/

使用方法如上圖所示,只需輸入模型結(jié)構(gòu)中各層的參數(shù)配置。

挺好用的不過它目標(biāo)分辨率太低了,放大之后不清晰,達(dá)不到印刷的需求。
這一個工具名叫draw_convnet,由Borealis公司的員工Gavin Weiguang Ding提供。

https://github.com/gwding/draw_convnet
看看畫的圖如下,核心工具是matplotlib,圖不酷炫,但是好在規(guī)規(guī)矩矩,可以嚴(yán)格控制,論文用挺合適的。

類似的工具還有:https://github.com/yu4u/convnet-drawer
下面要說的是這個,我最常用的,caffe的網(wǎng)絡(luò)結(jié)構(gòu)可視化工具,大名鼎鼎的netscope,由斯坦福AILab的Saumitro Dasgupta開發(fā),找不到照片就不放了,地址如下:
https://github.com/ethereon/netscope


左邊放配置文件,右邊出圖,非常方便進(jìn)行網(wǎng)絡(luò)參數(shù)的調(diào)整和可視化。這種方式好就好在各個網(wǎng)絡(luò)層之間的連接非常的方便。
再分享一個有意思的,不是畫什么正經(jīng)圖,但是把權(quán)重都畫出來了。
http://scs.ryerson.ca/~aharley/vis/conv/


看了這么多,有人已經(jīng)在偷偷笑了,上PPT呀,想要什么有什么,想怎么畫就怎么畫。


好消息,小白學(xué)視覺團(tuán)隊的知識星球開通啦,為了感謝大家的支持與厚愛,團(tuán)隊決定將價值149元的知識星球現(xiàn)時免費(fèi)加入。各位小伙伴們要抓住機(jī)會哦!

交流群
歡迎加入公眾號讀者群一起和同行交流,目前有SLAM、三維視覺、傳感器、自動駕駛、計算攝影、檢測、分割、識別、醫(yī)學(xué)影像、GAN、算法競賽等微信群(以后會逐漸細(xì)分),請掃描下面微信號加群,備注:”昵稱+學(xué)校/公司+研究方向“,例如:”張三 + 上海交大 + 視覺SLAM“。請按照格式備注,否則不予通過。添加成功后會根據(jù)研究方向邀請進(jìn)入相關(guān)微信群。請勿在群內(nèi)發(fā)送廣告,否則會請出群,謝謝理解~

