1. 问题描述
在定义transforms.Compose()时,把ToTensor()方法写在了resize的方法之前,造成顺序不对
2. 解决方案:
将ToTensor()方法移动到resize之后
将
trans = transforms.Compose([transforms.ToTensor(),
transforms.Resize(224)])
换成
trans = transforms.Compose([transforms.Resize(224),
transforms.ToTensor()])