numpy、list、pillow、tensor数据转换

numpy、list、pillow、tensor数据转换,第1张

  • 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)

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/922993.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-16
下一篇 2022-05-16

发表评论

登录后才能评论

评论列表(0条)

保存