pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
作者：小星星* 发布时间：2023-08-12 07:44:34
标签：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)
来源：https://blog.csdn.net/zsx1713366249/article/details/93619101
0
投稿猜你喜欢
- 在使用Ajax过程中，有时候总会遇到一些难题，浏览器兼容、编码、IE下的特殊处理等等，偶尔会搞的人头昏脑胀哭笑不得，这里列一些小贴士，或许有
- 而Easp类中提供了大量实用的ASP通用过程及方法，可以简化大部分的ASP操作。目前只提供了VBScript版，JScript版将来可能会提
- 内容摘要：图片随机显示是一个应用非常广泛的技巧。比如随机banner的显示，当你进入一个网站时它的banner总是不同的，或者总有内容不同的
- 1、简述asp的6个内置对象和其功能答案 : session ,server ,response,request,objectcontext
- 编者注：当讲到了性能优化和案例方面的东西，就要想到如何从开发人员的角度进行了理解，认识SQL是如何执行，以及如何学习高级的SQL，这篇文章对
- 目前计算机中用得最广泛的字符集及其编码，是由美国国家标准局(ANSI)制定的ASCII码（American Stand ard C
- 在SQL Server数据库中，主要是通过角色来继承相关的权限。但是，这个权限继承很容易造成权限上的冲突。如现在有个销售员账户SALE1，有
- 最近在改个程序用到了在js中设置css的float属性，以为和平常的写法一样，原来不是，只好去请教google，原来...首先大家先来看一下
- Oracle中有多种方法可以向数据库或服务器文件系统上载文件，这里主要介绍如下三种：Oracle HTTP Server（OHS）的mod_
今天在GOOGLE上查图片资料，这一幕真让我纠结啊：使用【向前】【向后】这种说法，就默认了有一个对比坐标，那就是当前显示的4张缩略图。点击【- 第一题：ASP中，VBScript的唯一的数据类型是什么？第二题：在ASP中，VBScript有多种控制程序流程语句，如If…Then, S
- 前言opencv中封装了一个专门用于求解cv::Mat均值的函数，即cv::mean(&cv::Mat)，该函数会得到Mat中各个通
- 其实不光是上面描述的情况会锁住表，还有很多种场景会使表放生死锁，解锁其实很简单，下面用一个示例来讲解： 1 首先创建一个测试用的表： 代码如
- 注：IE8以前的版本均不支持该特性为了向文档中插入生成内容，可以使用:before与:after伪元素。如，我想在所有链接的后面加上&quo
- 一、什么是 Postman（前世今生）Postman 诞生于 2013 年，一开始只是 Abhinav Asthana 着手于解决 API
- 在许多人看来，HTML应该是WEB制作所有语言中最简单的语言，因为它不需要编译、封闭等，甚至只需要一个记事本就可以让其在浏览器中呈现出来。所
- 一、首先我们来填个坑支付验签失败这个问题折磨了我两天，官方文档比较含糊不清。各种百度下来的方法试过之后也不尽人意，最后发现问题是没有二次签名
- hmac主要应用在身份验证中，它的使用方法是这样的：1. 客户端发出登录请求（假设是浏览器的GET请求）2. 服务器返回一个随机值，并在会话
- 问题：关于如何生成随机记录（二）如何从指定表中随机抽取一定量的记录？sql server 中 select top 10 * fr
- 第一列按照goodsid局部分组，然后在分组后的记录中按照audittime升序排序得到序号，从而显示某商品得第几次变迁。 第二列是取该商品
