AttributeError: module ‘torchvision.transforms’ has no attribute ‘Scale’
背景:
在使用transforms模型对图像预处理时,发现transforms没有Scale这个属性,原来是新版本中已经删除了Scale这个属性,改成Resize了
原因分析:
主要是torchvision的版本不一样,新版本的torchvision中的transforms没有Scale属性,改成Resize就好。
同理,如果没有Resize属性,可能是你安装了之前的版本,Resize改成Scale就好
解决方案:
Before:文章来源:https://www.toymoban.com/news/detail-519603.html
transform = transforms.Compose([transforms.Scale([128, 128]),transforms.ToTensor()])
After:文章来源地址https://www.toymoban.com/news/detail-519603.html
transform = transforms.Compose([transforms.Resize([128, 128]),transforms.ToTensor()])
到了这里,关于AttributeError: module ‘torchvision.transforms‘ has no attribute ‘Scale‘的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!