深度學習-逆卷積神經網絡ConvTranspose2d

        卷積神經網絡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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章