transforms.Resize(224) 和 transforms.Resize([224,224]) 有神魔区别

`transforms.Resize(224)` 和 `transforms.Resize([224,224])` 的区别在于输入的参数类型不同。

`transforms.Resize(224)` 中的参数 `224` 表示输出图像的较短边将被调整为 224 像素，而另一条边将按比例缩放。如果输入图像的宽为 w，高为 h，那么输出图像的大小将为 `(224, 224*h/w)` 或 `(224*w/h, 224)`，具体取决于哪个边是较短的。

而 `transforms.Resize([224,224])` 中的参数 `[224,224]` 表示输出图像的大小将调整为 224x224，即使输入图像的宽高比与输出图像的宽高比不同，也会进行强制缩放。这可能会导致图像的形状发生变化。

因此，如果希望保持输入图像的宽高比，请使用 `transforms.Resize(224)`；如果希望强制缩放图像到指定大小，请使用 `transforms.Resize([224,224])`。