TypeError_ an integer is required (got type tuple) RandomResizedCrop

TypeError: an integer is required (got type tuple) RandomResizedCrop

標籤(空格分隔): 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)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章