卷积神经网络图像卷积池化尺寸计算器
生活随笔
收集整理的這篇文章主要介紹了
卷积神经网络图像卷积池化尺寸计算器
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
卷積神經(jīng)網(wǎng)絡(luò)圖像尺寸計(jì)算器 (CNN_ImageSize_Calculator)
最近在研究圖片超分辨率的問題,發(fā)現(xiàn)大量計(jì)算圖像尺寸的問題。上網(wǎng)查找后發(fā)現(xiàn)網(wǎng)上大部分的神經(jīng)網(wǎng)絡(luò)可視化軟件,像 TensorboardX,Netron 等均是側(cè)重于可視化模型層中間的參數(shù)矩陣,并不太重視圖像尺寸的問題。剛好自己又會(huì)一點(diǎn) C# 可視化窗體程序的知識(shí),就簡單寫了一個(gè)計(jì)算器,我給起的名字是“卷積神經(jīng)網(wǎng)絡(luò)圖像尺寸計(jì)算器”。
可視化界面如下圖所示。
共包含了 16 層,每層有三種計(jì)算方式,包括 Conv2d,Pool2d,Deconv2d,和一種默認(rèn)不使用模式:Null,四種模式通過下拉菜單控制網(wǎng)絡(luò)的計(jì)算方式。
每層網(wǎng)絡(luò)下方的淺藍(lán)色 Width 和 Height 文字框,顯示每層網(wǎng)絡(luò)的輸出結(jié)果,也就是經(jīng)過此網(wǎng)絡(luò)模塊的處理后,圖像尺寸的變化。
源代碼已開源,感興趣的可移步至我的 Github 查看,fly吧
換了個(gè)配色
總結(jié)
以上是生活随笔為你收集整理的卷积神经网络图像卷积池化尺寸计算器的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Python 列表前加 *号
- 下一篇: PyTorch view() 函数