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

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 关于#单片机#的问题:以ATMEGA128或相近型号单片机为控制器设计直流电机调速的闭环控制系统(相关搜索:设计报告|软件设计|流程图)
  • ¥15 打开软件提示错误:failed to get wglChoosePixelFormatARB
  • ¥30 电脑误删了手机的照片怎么恢复?
  • ¥15 (标签-python|关键词-char)
  • ¥15 python+selenium,在新增时弹出了一个输入框
  • ¥15 苹果验机结果的api接口哪里有??单次调用1毛钱及以下。
  • ¥20 学生成绩管理系统设计
  • ¥15 来一个cc穿盾脚本开发者
  • ¥15 CST2023安装报错
  • ¥15 使用diffusionbert生成文字 结果是PAD和UNK怎么办