PyTorch载入图片后ToTensor解读
作者:互联网
转:https://www.cnblogs.com/ocean1100/p/9494640.html
PyTorch在做一般的深度学习图像处理任务时,先使用dataset类和dataloader类读入图片,在读入的时候需要做transform变换,其中transform一般都需要ToTensor()操作,将dataset类中__getitem__()方法内读入的PIL或CV的图像数据转换为torch.FloatTensor。
标签:__,载入,transform,dataset,ToTensor,PyTorch,读入 来源: https://blog.csdn.net/ZhuLT23/article/details/88846817