今天的项目有这么一个调用:
from torchvision.transforms import Compose, CenterCrop, Scale
运行之后报错:
ImportError: cannot import name ‘Scale’ from ‘torchvision.transforms’
经查阅,torchvision的版本不一样,新版本的torchvision中的transforms没有Scale属性。考虑到这是一个比较老的项目了,应该是这个原因。
因此,直接把Scale 改成Resize。
from torchvision.transforms import Compose, CenterCrop, Resize
再把程序中的Scale全改成Resize, 最后运行,成功。