其他分享
首页 > 其他分享> > PyTorch载入图片后ToTensor解读

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