图像插值算法

最新推荐文章于 2024-05-08 11:25:05 发布

Jokic_Rn

最新推荐文章于 2024-05-08 11:25:05 发布

阅读量712

点赞数 1

文章标签： pytorch

本文链接：https://blog.youkuaiyun.com/weixin_44823313/article/details/126505334

版权

本文介绍在使用PyTorch进行图像变换时，如何处理警告'interpolation should be of type InterpolationMode'，并详细讲解了最近邻、双线性和Lanczos插值等。未来版本将弃用数字参数，建议采用官方提供的InterpolationMode枚举值进行转换。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

图像插值算法

最近使用pytorch对图像进行变换遇到一个
warning:"Argument interpolation should be of type InterpolationMode instead of int. "
也就是这个函数

transforms.Resize(self.height, self.width , interpolation=1)

这个warning就是说参数不能在传数字了了，应该填入InterpolationMode ，查看源码，torchvision说下一个版本更新，会弃用数字参数，应该填入枚举值。目前是兼容两种写法，官方转化数字成枚举值的代码如下：

def _interpolation_modes_from_int(i: int) -> InterpolationMode:
	inverse_modes_mapping = {
   	0: InterpolationMode.NEAREST,
   	2: InterpolationMode.BILINEAR,
   	3: InterpolationMode.BICUBIC,
    4: InterpolationMode.BOX,
   	5: InterpolationMode.HAMMING,
   	1: InterpolationMode.LANCZOS,        
   	}
    return inverse_modes_mapping[i]