nn.ReflectionPad2d(鏡像填充)


1、官方文檔上 是這個     

 

 

2、輸出是   

H(out) = H(in) + paddingTop + paddingBottom
W(out) = W(in) + paddingLeft + paddingRight 
直接來例子試試    最重要的是下面 # laft, right, top, bottom  這個注釋

input = torch.randn(64, 3, 220, 220) # input size

# 輸入4-tuple
pad = nn.ReflectionPad2d((3, 3, 5, 5))     # laft, right, top, bottom
output = pad(input) # size(64, 3, 230, 226)
# 輸入int
pad = nn.ReflectionPad2d(3)
output = pad(input) # size(64, 3, 226, 226)
通過運算 發現的確是這樣的結果      所以記住公式就行了


轉自:https://blog.csdn.net/bc521bc/article/details/85134539


免責聲明!

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



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