pytorch 中pad函数toch.nn.functional.pad()的用法


Posted in Python onJanuary 08, 2020

padding操作是给图像外围加像素点。

为了实际说明操作过程,这里我们使用一张实际的图片来做一下处理。

pytorch 中pad函数toch.nn.functional.pad()的用法

这张图片是大小是(256,256),使用pad来给它加上一个黑色的边框。具体代码如下:

import torch.nn,functional as F
import torch
from PIL import Image
im=Image.open("heibai.jpg",'r')

X=torch.Tensor(np.asarray(im))
print("shape:",X.shape)
dim=(10,10,10,10)
X=F.pad(X,dim,"constant",value=0)

padX=X.data.numpy()
padim=Image.fromarray(padX)
padim=padim.convert("RGB")#这里必须转为RGB不然会

padim.save("padded.jpg","jpeg")
padim.show()
print("shape:",padX.shape)

输出:

shape: torch.Size([256, 256])
shape: (276, 276)

可以看出给原图四个方向给加上10维度的0,维度变为256+10+10得到的图像如下:

pytorch 中pad函数toch.nn.functional.pad()的用法

我们在举几个简单例子:

x=np.asarray([[[1,2],[1,2]]])
X=torch.Tensor(x)
print(X.shape)
pad_dims = (
          2, 2,
          2, 2,
          1, 1,

        )
X=F.pad(X,pad_dims,"constant")
print(X.shape)
print(X)

输出:

torch.Size([1, 2, 2])
torch.Size([3, 6, 6])
tensor([[[ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.]],

    [[ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 1., 2., 0., 0.],
     [ 0., 0., 1., 2., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.]],

    [[ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.]]])

可以知若pid_sim为(2,2,2,2,1,1)则原维度变化是2+2+2=6,1+1+1=3.也就是第一个(2,2) pad的是最后一个维度,第二个(2,2)pad是倒数第二个维度,第三个(1,1)pad是第一个维度。

再举一个四维度的,但是只pad三个维度:

x=np.asarray([[[[1,2],[1,2]]]])
X=torch.Tensor(x)#(1,2,2)
print(X.shape)
pad_dims = (
          2, 2,
          2, 2,
          1, 1,

        )
X=F.pad(X,pad_dims,"constant")#(1,1,12,12)
print(X.shape)
print(X)

输出:

torch.Size([1, 1, 2, 2])
torch.Size([1, 3, 6, 6])
tensor([[[[ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.]],

     [[ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 1., 2., 0., 0.],
     [ 0., 0., 1., 2., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.]],

     [[ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.]]]])

再举一个四维度的,pad四个维度:

x=np.asarray([[[[1,2],[1,2]]]])
X=torch.Tensor(x)#(1,2,2)
print(X.shape)
pad_dims = (
          2, 2,
          2, 2,
          1, 1,
          2, 2

        )
X=F.pad(X,pad_dims,"constant")#(1,1,12,12)
print(X.shape)
print(X)

输出:

torch.Size([1, 1, 2, 2])
torch.Size([5, 3, 6, 6])
tensor([[[[ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.]],

     [[ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.]],

     [[ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.],
     [ 0., 0., 0., 0., 0., 0.]]],


.........太多了

以上这篇pytorch 中pad函数toch.nn.functional.pad()的用法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python批量同步web服务器代码核心程序
Sep 01 Python
详解Python中的join()函数的用法
Apr 07 Python
连接Python程序与MySQL的教程
Apr 29 Python
在Python中操作文件之read()方法的使用教程
May 24 Python
Python中的连接符(+、+=)示例详解
Jan 13 Python
python抓取网页中链接的静态图片
Jan 29 Python
python脚本监控Tomcat服务器的方法
Jul 06 Python
详解python使用turtle库来画一朵花
Mar 21 Python
python中selenium操作下拉滚动条的几种方法汇总
Jul 14 Python
基于python求两个列表的并集.交集.差集
Feb 10 Python
python数据预处理 :数据共线性处理详解
Feb 24 Python
关于Python不换行输出和不换行输出end=““不显示的问题(亲测已解决)
Oct 27 Python
pycharm激活码有效到2020年11月底
Sep 18 #Python
pymysql的简单封装代码实例
Jan 08 #Python
pytorch使用 to 进行类型转换方式
Jan 08 #Python
Pycharm 2020年最新激活码(亲测有效)
Sep 18 #Python
Pytorch to(device)用法
Jan 08 #Python
pycharm 2019 最新激活方式(pycharm破解、激活)
Sep 22 #Python
Python语言异常处理测试过程解析
Jan 08 #Python
You might like
php项目打包方法
2008/02/18 PHP
利用php+mysql来做一个功能强大的在线计算器
2010/10/12 PHP
PHP批量去除BOM头代码分享
2015/06/26 PHP
php+ajax无刷新上传图片的实现方法
2016/12/06 PHP
JavaScript实现相册弹窗功能(zepto.js)
2016/06/21 Javascript
jQuery子元素过滤选择器用法示例
2016/09/09 Javascript
Bootstrap DateTime Picker日历控件简单应用
2017/03/25 Javascript
nodejs使用express获取get和post传值及session验证的方法
2017/11/09 NodeJs
vue devtools的安装与使用教程
2018/08/08 Javascript
详解webpack打包nodejs项目(前端代码)
2018/09/19 NodeJs
ES6 fetch函数与后台交互实现
2018/11/14 Javascript
elementUI中Table表格问题的解决方法
2018/12/04 Javascript
浅谈Vue.js中如何实现自定义下拉菜单指令
2019/01/06 Javascript
Vue中实现权限控制的方法示例
2019/06/07 Javascript
jquery多级树形下拉菜单的实例代码
2019/07/09 jQuery
vue-loader中引入模板预处理器的实现
2019/09/04 Javascript
详解vue 组件的实现原理
2020/11/12 Javascript
python实现apahce网站日志分析示例
2014/04/02 Python
python连接字符串的方法小结
2015/07/13 Python
python+selenium实现登录账户后自动点击的示例
2017/12/22 Python
python实现远程控制电脑
2019/05/23 Python
python 计算数据偏差和峰度的方法
2019/06/29 Python
Window系统下Python如何安装OpenCV库
2020/03/05 Python
jupyter修改文件名方式(TensorFlow)
2020/04/21 Python
解决Pytorch自定义层出现多Variable共享内存错误问题
2020/06/28 Python
Python自动登录QQ的实现示例
2020/08/28 Python
详解Python 函数参数的拆解
2020/09/02 Python
Python实现LR1文法的完整实例代码
2020/10/25 Python
css3教程之倾斜页面
2014/01/27 HTML / CSS
店长岗位职责
2013/11/21 职场文书
汉语言文学毕业生自荐信范文
2014/03/24 职场文书
个人创业事迹材料
2014/12/30 职场文书
公司员工违法违章行为检讨书
2019/06/24 职场文书
OpenCV-Python实现图像平滑处理操作
2021/06/08 Python
python编程项目中线上问题排查与解决
2021/11/01 Python
Github 使用python对copilot做些简单使用测试
2022/04/14 Python