TypeError: tensor is not a torch image.

最新推荐文章于 2023-02-20 17:27:49 发布

烂桃先生之前的账号

最新推荐文章于 2023-02-20 17:27:49 发布

阅读量3.7k

点赞数 9

分类专栏： Pytorch

本文链接：https://blog.csdn.net/weixin_42464187/article/details/104794983

版权

Pytorch 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

运行写好的dataLoader类报错如下
TypeError: tensor is not a torch image.

原因是transforms.Compose里面函数的顺序问题导致的，normalize需要在ToTensor的后面。

错误的顺序：

self.normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], 
                                      std=[0.229, 0.224, 0.225])

self.data_transform = transforms.Compose([self.normalize,
                                          transforms.ToTensor(),
                                          transforms.ColorJitter(),])

正确的顺序：

self.normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], 
                                      std=[0.229, 0.224, 0.225])

self.data_transform = transforms.Compose([transforms.ToTensor(),
					                      self.normalize,
										  transforms.ColorJitter(),])