numpy、list、pillow、tensor数据转换

PIL不自带图片浏览工具,会调用你系统的图片浏览工具
plt.imshow()函数负责对图像进行处理,并显示其格式,而plt.show()则是将plt.imshow()处理后的函数显示出来。

tensor转PIL转numpy转cv2

# crop_image为模型输出特征层tensor
image = cv2.cvtColor(np.asarray(ToPILImage(crop_image.cpu())), cv2.COLOR_RGB2BGR)
  • np.array和np.asarray区别
    对于非ndarray(如输入是list列表),两者无任何区别。都将输入数据转化为ndarray,并且占用新的内存,不跟输入数据占一块内存。
    对于ndarray,array仍然会分配新的内存,而asarray跟输入数据指向同一个内存,换言之,如果输入数据发生改变,asarray也会发生改变,而array还是原来的数据。

  • numpy和pillow

im = np.array(Image.open(im))
im = Image.fromarray(im)
  • list和numpy互转
temp = np.array(list) 
arr = temp.tolist() 
  • numpy和tensor互转
    注意,第一只有都在CPU上的才能相互转换
    第二直接使用from_numpy的是指向了同一个地址
    第三如果想创建新的b = torch.tensor(a)
import torch
import numpy as np

a = np.array([1, 2, 3])
b = torch.from_numpy(a)
# c=b.numpy()
d = torch.tensor(a)
e = torch.as_tensor(a)
a[1] = 3
print(b)
print(d)
print(e)

输出:
tensor([1, 3, 3], dtype=torch.int32)
tensor([1, 2, 3], dtype=torch.int32)
tensor([1, 3, 3], dtype=torch.int32)
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值