明明就是PIL为什么又说是tensor呢?这种情况是因为顺序不对
注意 transforms.ToTensor(),的位置
例如
错误顺序
transform = transforms.Compose(
[
transforms.ToTensor(),
transforms.Resize(224),
transforms.CenterCrop(224),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]
)
正确顺序
transform = transforms.Compose(
[
transforms.Resize(224),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]
)