PyTorch载入图片后ToTensor解读

转:https://www.cnblogs.com/ocean1100/p/9494640.html

PyTorch在做一般的深度学习图像处理任务时,先使用dataset类和dataloader类读入图片,在读入的时候需要做transform变换,其中transform一般都需要ToTensor()操作,将dataset类中__getitem__()方法内读入的PIL或CV的图像数据转换为torch.FloatTensor。

你可能感兴趣的:(PyTorch载入图片后ToTensor解读)