【總結】深度學習圖片卷積輸出大小計算公式


原文鏈接:https://www.jianshu.com/p/c56a37093cfa

輸入圖片經過卷積后所得特征圖大小的計算公式

先定義幾個參數

  • 輸入圖片大小 W×W
  • Filter大小 F×F
  • 步長 S
  • padding的像素數 P

於是我們可以得出

N = (W − F + 2P )/S+1

輸出圖片大小為 N×N

如:輸入圖片的shape為[10,3,227,227],對應輸入圖片大小為227x227

輸出num_output為96,對應通道數

Filter大小(kernel_size)為11,對應為11x11

步長S(stride)為4

N = (W − F + 2P )/S+1 = (227-11+2x0)/4 + 1 = 55

則輸出的特征圖大小shape為[10,96,55,55].

多說一句:卷積核的個數等於輸出特征圖的通道數。(這個一般都不會明確說出卷積核的個數)

 

反卷積得到的圖片大小計算方式:

反卷積的大小是由卷積核大小與滑動步長決定, in是輸入大小, k是卷積核大小, s是滑動步長, out是輸出大小
得到 out = (in - 1) * s + k

例如 輸入:2x2, 卷積核:4x4, 滑動步長:3, 輸出:7x7 ,其計算過程就是, (2 - 1) * 3 + 4 = 7

 

2018-04-06:看到轉置卷積中存在指定padding的情況,如何計算特征圖的大小。(假設padding為p)

out = (in - 1)* s - 2 * p + k (與上式不同的是需要減去兩倍的padding)

 

池化得到的特征圖大小計算方式:

卷積向下取整,池化向上取整。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM