torchvision将图片转换成Tensor：transforms.ToTensor()

最新推荐文章于 2025-03-08 16:30:30 发布

追光少年呀

最新推荐文章于 2025-03-08 16:30:30 发布

阅读量2.6w

点赞数 29

分类专栏： Pytorch数据载入文章标签： Pytorch 数据集载入

本文链接：https://blog.csdn.net/aLWX_hust/article/details/86591783

版权

transforms.ToTensor()
（1） transforms.ToTensor() 将numpy的ndarray或PIL.Image读的图片转换成形状为(C,H, W)的Tensor格式，且/255归一化到[0,1.0]之间
（2）通道的具体顺序与cv2读的还是PIL.Image读的图片有关系
cv2:(B,G,R)
PIL.Image:(R, G, B)

import torch
import cv2
from PIL import Image
from torchvision import transforms
image = cv2.imread('myimage.jpg')  # numpy数组格式（H，W，C=3），通道顺序（B,G,R)
image2 = Image