Pytorch中的variable, tensor与numpy相互转化的方法
2018-03-06 16:58
513 查看
在使用pytorch作为深度学习的框架时,经常会遇到变量variable、张量tensor与矩阵numpy的类型的相互转化的问题,本章结合这实际图像对此转化方法进行实现。
1.加载需要用到的模块import torch
from torch.autograd import Variable
import matplotlib.pyplot as plt
import matplotlib.image as mpimg2.显示图片与图片中的一部分区域test_img = mpimg.imread('example1.jpg')
i_x = 20
i_y = 85
sub_img = test_img[i_y:i_y + 100,i_x:i_x + 100,:] #numpy类型
3.将numpy矩阵转换为Tensor张量sub_ts = torch.from_numpy(sub_img) #sub_img为numpy类型4.将Tensor张量转化为numpy矩阵sub_np1 = sub_ts.numpy() #sub_ts为tensor张量5.将numpy转换为Variablesub_va = Variable(torch.from_numpy(sub_img))6.将Variable张量转化为numpysub_np2 = sub_va.data.numpy()
1.加载需要用到的模块import torch
from torch.autograd import Variable
import matplotlib.pyplot as plt
import matplotlib.image as mpimg2.显示图片与图片中的一部分区域test_img = mpimg.imread('example1.jpg')
i_x = 20
i_y = 85
sub_img = test_img[i_y:i_y + 100,i_x:i_x + 100,:] #numpy类型
3.将numpy矩阵转换为Tensor张量sub_ts = torch.from_numpy(sub_img) #sub_img为numpy类型4.将Tensor张量转化为numpy矩阵sub_np1 = sub_ts.numpy() #sub_ts为tensor张量5.将numpy转换为Variablesub_va = Variable(torch.from_numpy(sub_img))6.将Variable张量转化为numpysub_np2 = sub_va.data.numpy()
相关文章推荐
- 浅谈pytorch和Numpy的区别以及相互转换方法
- datatable与实体类之间相互转化的几种方法
- UTF-8 和 GBK 的 NSString 相互转化的方法
- iOS中时间与时间戳的相互转化(iOS10中 其他方法好像没什么用呀)
- c++的字符串char与string相互转化,以及string的,切割,替换字符,字符串拼接方法(object向c++转型系列教程2)
- pytorch: tensor类型的构建与相互转换实例
- UTF-8 和 GBK 的 NSString 相互转化的方法
- Javabean和map相互转化方法代码示例
- 关于文档或pdf或网页和数据库的相互转化的方法
- UTF-8 和 GBK 的 NSString 相互转化的方法
- Numpy中矩阵matrix读取一列的方法及数组和矩阵的相互转换实例
- python字符串str和字节数组相互转化方法
- 对象与Byte数组相互转化工具方法
- 【Java】字符串变量和整形之间的相互转化与静态方法static
- ios开发中object-c中UTF-8 和 GBK 的 NSString 相互转化的方法
- numpy的array和python中自带的list之间相互转化
- 用list去初始化numpy的array数组 numpy的array和python中自带的list之间相互转化
- 总结Java标准类库中类型相互转化的方法
- UTF-8 和 GBK 的 NSString 相互转化的方法
- numpy的array和python中自带的list之间相互转化