pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
更新时间：2020年01月13日 09:08:13 作者：小星星*
今天小编就为大家分享一篇pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
1, 创建pytorch 的Tensor张量：
torch.rand((3,224,224)) #创建随机值的三维张量，大小为（3,224,224） torch.Tensor([3,2]) #创建张量，[3,2]
2, cpu上的tensor和GPU即pytorch创建的tensor的相互转化
b = a.cpu() # GPU → CPU a = b.cuda() #CPU → GPU
3, tensor和numpy的转化
b = a.numpy() # tensor转化为 numpy数组 a = b.from_numpy() # numpy数组转化为tensor
4, torch的GPU tensor保存为图片
import scipy.misc scipy.misc.imsave(‘pic_name',img) #img为二维张量，比如（224,224），保存为黑白图
5, 堆叠矩阵，形成彩色图片
img = np.stack((ia,b,c),dim) #堆叠矩阵a,b,c 可用于三通道图像的保存 dim表示要增加的维度， #比如a,b,c均为（224,224）大小的矩阵，那么令dim=-1，则 img的维度为（224,224,3)
6, 从numpy数组保存图片
from PIL import Image
im = Image.fromarray(A)
im.save("your_file.jpeg")
7, 读取图片为矩阵:
import matplotlib.image
im = matplotlib.image.imread('0_0.jpg')
8, 保存矩阵为图片:
import numpy as np
import scipy.misc
x = np.random.random((600,800,3))
scipy.misc.imsave('meelo.jpg', x)
以上这篇pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
Python3加密解密库Crypto的RSA加解密和签名/验签实现方法实例
这篇文章主要介绍了Python3加密解密库Crypto的RSA加解密和签名/验签实现方法实例,需要的朋友可以参考下2020-02-02
详解Python3.8+PyQt5+pyqt5-tools+Pycharm配置详细教程
这篇文章主要介绍了Python3.8+PyQt5+pyqt5-tools+Pycharm配置详细教程，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友参考下吧2020-11-11
今天小编就为大家分享一篇python 字符串追加实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
这篇文章主要继续上一篇文章100 个 Python 小例子（练习题一）来完成100 个 Python 小例子，本文包括字母识词、反向输出II、表转字符串、设置输出颜色、算素数等例子,需要的朋友可以参考一下2021-10-10
这篇文章主要介绍了在Python的Django框架中包装视图函数的方法,即requires_login的相关方法,需要的朋友可以参考下2015-07-07
前段时间在微博看到一段摸鱼人的倒计时模板，感觉很有意思，于是我用了一个小时的时间写一个页面出来，下面小编把实现过程分享给大家，对Python摸鱼倒计时界面感兴趣的朋友一起看看吧2021-12-12
这篇文章主要介绍了如何用C代码给Python写扩展库(Cython)，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2019-05-05
这篇文章主要介绍了Python使用三种方法实现PCA算法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-12-12
这篇文章主要为大家详细介绍了如何通过Python语言实现视频分帧的功能，文中的示例代码讲解详细，感兴趣的小伙伴可以跟随小编一起尝试一下2023-03-03
这篇文章主要介绍了Python中的切片函数Slice详解,切片是对有序的集合而言,意思从有序集合里提取数据构成子集集合,给定提取的起点start、终点end以及方向上的步长step,能否切出非空子集,起点start需能沿步长方向上到达终点,需要的朋友可以参考下2023-09-09
最新评论