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