解决 UserWarning: Argument interpolation should be of type InterpolationMode instead of int.

问题描述

运行代码出现了如下报错

C:\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\torchvision\transforms\transforms.py:288: 
UserWarning: Argument interpolation should be of type InterpolationMode instead of int. Please, use InterpolationMode enum.

有博客提到是当前 torchvision 和 pillow 版本不匹配造成的。

解决办法

修改相应的代码，根据提示使用 use InterpolationMode enum.，具体可以参考 pytorch 官网

1. 导入InterpolationMode，from torchvision.transforms import InterpolationMode
2. 将代码中 interpolation 后的参数换成 InterpolationMode.BICUBIC，参数可以是 Image.BICUBIC ，也可以是整数，如下面的 3

例如：

transforms.Resize(size=(opt.h, opt.w),interpolation=3), #Image.BICUBIC

改为

transforms.Resize((opt.h, opt.w), interpolation=InterpolationMode.BICUBIC),

参考链接

1. UserWarning: Argument interpolation should be of type InterpolationMode instead of int 解决方案
2. pytorch 官网
3. 【pytroch warning】“Argument interpolation should be of type InterpolationMode instead of int. “