标签(空格分隔): pytorch
这个错误,可能是因为版本引起的,但是我在网上没有找到解决方案,自己查看源码发现了问题。
代码如下:
from PIL import Image img = Image.open("face-265.bmp").convert('RGB') transforms.RandomResizedCrop((128,128), scale=(0.9, 1), ratio=(1, 1))(img)RandomResizedCrop这个Transform出错了,原因是输入是一个tuple,但这段代码别人能运行,所以应该版本原因,所以我去RandomResizedCrop里面查看了一下,发现:
def __init__(self, size, scale=(0.08, 1.0), ratio=(3. / 4., 4. / 3.), interpolation=Image.BILINEAR): self.size = (size, size) self.interpolation = interpolation self.scale = scale self.ratio = ratio所以它的输入size应该是一个值,而不是一个tuple,所以代码改为:
from PIL import Image img = Image.open("face-265.bmp").convert('RGB') transforms.RandomResizedCrop(128, scale=(0.9, 1), ratio=(1, 1))(img)