分类：pytorch中nn.ZeroPad2d()

2021
04-27

pytorch中的nn.ZeroPad2d()零填充函数实例详解

在卷积神经网络中，有使用设置padding的参数，配合卷积步长，可以使得卷积后的特征图尺寸大小不发生改变，那么在手动实现图片或特征图的边界零填充时，常用的函数是nn.ZeroPad2d(),可以指定tensor的四个方向上的填充，比如左边添加1dim、右边添加2dim、上边添加3dim、下边添加4dim，即指定paddin参数为（1，2，3，4），本文中代码设置的是（3，4，5，6）如下：importtorch.nnasnnimportcv2importtorchvisionfromtorchvision.... 继续阅读 >