Resize()和RandomHorizontalFlip()等方法要求输入的图像为PIL Image,而正则化操作Normalize()处理的是tensor格式的图像数据。
因此totensor()应放在transforms.resize()等之后,transforms.normalize()之前。且输入PIL.Image图像
然而,当你对 torch Tensor 对象使用 transforms.Resize(256) 时,通常不会出现问题的原因是:在 PyTorch 中,图像通常首先被加载为 PIL Image 对象,然后转换为 torch Tensor 对象。因此,如果你的输入是 torch Tensor,它实际上是从 PIL Image 转换而来的,所以可以正常使用 transforms.Resize(256)
transforms中totensor应该放在哪里?
于 2024-06-13 15:15:10 首次发布