torch 笔记

torchvision

transforms

from torchvision import transforms

transform = transforms.Compose([
        transforms.ToTensor(),
    transforms.Normalize(mean = (0.5, 0.5, 0.5), std = (0.5, 0.5, 0.5)),
])

ToTensor() 将 PIL.Image 或 numpy.ndarray 数据进转化为 torch.FloadTensor，并归一化到[0, 1.0]。

  • 形状为 [H, W, C] 的 numpy.ndarray，转换成形状为 [C, H, W]，取值范围是 [0, 1.0] 的torch.FloadTensor。
  • 取值范围为 [0, 255] 的 PIL.Image，转换成形状为 [C, H, W]，取值范围是 [0, 1.0] 的torch.FloadTensor
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。