torchvision.transforms.Resize()函数解读

最新推荐文章于 2025-03-27 11:40:57 发布

原创最新推荐文章于 2025-03-27 11:40:57 发布

· 4.4w 阅读

16 ·

版权

本文介绍了一个用于调整PILImage对象大小的函数，该函数可以将图像缩放到指定的尺寸，支持按比例缩放，并提供了多种插值选项以优化图像质量。

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

函数作用
对于PIL Image对象进行resize的运算。
函数源码

class Resize(object):
    """Resize the input PIL Image to the given size.

    Args:
        size (sequence or int): Desired output size. If size is a sequence like
            (h, w), output size will be matched to this. If size is an int,
            smaller edge of the image will be matched to this number.
            i.e, if height > width, then image will be rescaled to
            (size * height / width, size)
        interpolation (int, optional): Desired interpolation. Default is
            ``PIL.Image.BILINEAR``
    """

    def __init__(self, size, interpolation=Image.BILINEAR):
        assert isinstance(size, int) or (isinstance(size, Iterable) and len(size) == 2)
        self.size = size
        self.interpolation = interpolation

    def __call__(self, img):
        """
        Args:
            img (PIL Image): Image to be scaled.

        Returns:
            PIL Image: Rescaled image.
        """
        return F.resize(img, self.size, self.interpolation)

    def __repr__(self):
        interpolate_str = _pil_interpolation_to_str[self.interpolation]
        return self.__class__.__name__ + '(size={0}, interpolation={1})'.format(self.size, interpolate_str)

参考链接