Ubutun20.04 Pycharm pytoch AttributeError: module ‘torchvision.transforms‘ has no attribute ‘Scale‘

在运行代码时遇到AttributeError,原因是torchvision的新版本中将Scale方法替换为了Resize。为解决此问题,只需将代码中的Scale替换为Resize,并保持其他参数不变,即可成功执行图像预处理步骤。
摘要由CSDN通过智能技术生成

运行代码后出现下面问题

  File "/home/sys120-1/cy/I2V-GAN-main/data/custom_dataset_data_loader.py", line 27, in initialize
    self.dataset = CreateDataset(opt)
  File "/home/sys120-1/cy/I2V-GAN-main/data/custom_dataset_data_loader.py", line 17, in CreateDataset
    dataset.initialize(opt)
  File "/home/sys120-1/cy/I2V-GAN-main/data/unaligned_dataset.py", line 23, in initialize
    self.transform = get_transform(opt)
  File "/home/sys120-1/cy/I2V-GAN-main/data/base_dataset.py", line 19, in get_transform
    transform_list.append(transforms.Scale(osize, Image.BICUBIC))
AttributeError: module 'torchvision.transforms' has no attribute 'Scale'

分析:在使用transforms模型对图像预处理时,发现transforms没有Scale这个属性
原因:主要是torchvision的版本不一样,新版本的torchvision中的transforms没有Scale属性,改成Resize了。
解决方案

transform_list.append(transforms.Scale(osize, Image.BICUBIC))

改成

transform_list.append(transforms.Resize(osize, Image.BICUBIC))

即可运行成功

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值