卷積神經網絡Conv2d是對圖片進行下采樣,圖片越來越小,而逆卷積神經網絡ConvTranspose2d是對圖片進行上採樣,圖片越來越大,卷積與逆卷積是互相對應的。
首先,不管stride等於幾,卷積公式都爲,逆卷積公式爲(注意這個p和s是對應的卷積操作的p和s)。
1.當stride=1時:
卷積操作:
由圖片得知,i=4,o=2,k=3,s=1,p=0,2=(4-3+2*0)/1+1
對應的逆卷積操作(只需要padding):
由圖片得知,i=2,o=4,k=3,s=1,p=0,4=(2-1)-2*0+3
2.當stride=2時:
卷積操作:
由圖片得知,i=5,o=3,k=3,s=2,p=1,3=(5-3+2*1)/2+1
對應逆卷積操作(不僅需要padding,還需要插值,即每個特徵圖中每個像素之間插入s-1個0):
由圖片得知,i=3,o=5,k=3,s=2,p=1,5=(3-1)*2-2*1+3