pytorch padding=’SAME’解决方法

  • Post author:
  • Post category:其他


tensorflow中的conv2有padding=‘SAME’这个参数。吴恩达讲课中说到当padding=(f-1)/2(f为卷积核大小)时则是SAME策略。但是这个没有考虑到空洞卷积的情况,也没有考虑到strides的情况。

查阅资料后发现网上方法比较麻烦。

手算,实验了一个早上,终于初步解决了问题。

分为两步:



填充多少

中文文档中有计算公式:

输入:




( N , C i n , H i n , W i n ) (N,C_in,H_in,W_in)






(


N


,





C










i


















n


,





H










i


















n






W










i


















n


)







输出:




( N , C o u t , H o u t , W o u t ) (N,C_out,H_out,W_out)






(







版权声明:本文为BenjaminYoung29原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。