torchvision.transforms.ToTensor()与torchvision.transforms.Compose()

torchvision.transforms.ToTensor()

        该函数可以将cv2读取的图片(图片的shape为(H x W xC)且数值范围为0~255)或者PIL.Image读取的图片转换为shape为(C x H xW )且数值范围为0~1的float tensor。

        其中有一点需要注意的是,cv2图片读取的通道顺序为(B,G,R),PIL.Image图片读取的通道顺序为(R,G,B)

torchvision.transforms.Compose()

torchvision.transforms.Compose(transforms),该函数就是将几个transforms组合在一起。

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

上述代码即将两个transforms组合到了transform里面,该思想与torch.nn.Sequential类似。
 

你可能感兴趣的:(计算机视觉,python,深度学习)