运行代码后出现下面问题
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))
即可运行成功