PILImage 和 Tensor转换 TypeError: Input type int64 is not supported

镇长1998

于 2020-11-24 22:32:12 发布

阅读量6.3k

点赞数 1

分类专栏：实验室

本文链接：https://blog.youkuaiyun.com/weixin_41514525/article/details/110086894

版权

实验室专栏收录该内容

41 篇文章

订阅专栏

# 报错代码

import torch
import torchvision
from PIL import Image

img = Image.open('./post_disaster.png')

tensor = torchvision.transforms.ToTensor()(img)      # 转换为Tensor

tensor = tensor.to(torch.long)

img = torchvision.transforms.ToPILImage()(tensor)  

img.show()

出错原因: transform.ToPILImage()(tensor) 该方法接受的tensor需要是torch.float32类型的参数, 不能是torch.int类型的参数, 修改tensor的dtype类型为torch.float32, 否则会报错如下图