木叶清风666 2022-04-06 14:10 采纳率: 0%
浏览 381

pytorch深度学习中torchvision.transforms.ToTensor()使用问题

问题遇到的现象和发生背景

关于torchvision.transforms.ToTensor(),官方解释是把一个取值范围是[0,255]的PIL.Image或者shape为(H,W,C)的numpy.ndarray,转换成形状为[C,H,W],取值范围是[0,1.0]的torch.FloadTensor,但是在使用过程中却发现只是将数据转化成了tensor,并没有映射到0~1

问题相关代码,请勿粘贴截图

from torchvision import transforms
import numpy as np
data = np.random.randint(0, 255, size=300)
img = data.reshape(10,10,3)
print(img.shape)
img_tensor = transforms.ToTensor()(img) # 转换成tensor
print(img_tensor)
输出见下图所示

运行结果及报错内容

结果并未将数据映射到0~1,进行归一化处理

img

  • 写回答

2条回答 默认 最新

  • Love And Program 人工智能领域新星创作者 2022-04-06 14:39
    关注

    只有Ndarray是np.uint8 类型的时候,才会归一化

    img


    有用记得采纳哦

    评论 编辑记录

报告相同问题?

问题事件

  • 创建了问题 4月6日

悬赏问题

  • ¥15 想用adb命令做一个通话软件,播放录音
  • ¥30 Pytorch深度学习服务器跑不通问题解决?
  • ¥15 部分客户订单定位有误的问题
  • ¥15 如何在maya程序中利用python编写领子和褶裥的模型的方法
  • ¥15 Linux权限管理相关操作(求解答)
  • ¥15 Bug traq 数据包 大概什么价
  • ¥15 在anaconda上pytorch和paddle paddle下载报错
  • ¥25 自动填写QQ腾讯文档收集表
  • ¥15 DbVisualizer Pro 12.0.7 sql commander光标错位 显示位置与实际不符
  • ¥15 android 打包报错