nn.ConvTranspose2d中output_padding參數理解

比如圖片尺寸6×6,步長爲2,卷積核爲3×3,padding=1,輸出的圖片尺寸應爲(6+2-3)/2+1 = 3。如果輸入圖片是5×5,步長爲2,卷積核爲3×3,padding=1,那麼輸出爲(5+2-3)/2+1=3,也是3*3。 這樣的話使用反捲積還原圖像尺寸時就會爭議。 解決爭議的辦法就是使用output_padding參數,output_padding的值默認爲stride-1,這樣還原
相關文章
相關標籤/搜索