RandomCrop的get_params函数中判断逻辑有问题？

最新推荐文章于 2024-11-27 20:44:51 发布

qq_30615001

最新推荐文章于 2024-11-27 20:44:51 发布

阅读量712

点赞数

文章标签： python pytorch

本文链接：https://blog.csdn.net/qq_30615001/article/details/129708388

版权

在torchvision的RandomCrop模块中，源码可能存在一个潜在问题。当输入图像尺寸小于预期的输出尺寸时，get_params函数会检查并抛出ValueError。然而，在特定情况下（h=th-1），异常检查可能不充分，导致torch.randint在之后的步骤报错。若在调用RandomCrop前未进行pad操作，可能会触发这个错误。

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

环境：

python3.8 torch1.12.1 torchvision0.13.1

在RandomCrop源码中，crop之前用了get_params函数获取crop的入参，防止crop出错。

但我在看get_params源码过程中发现，抛异常的条件应该是：if h < th or w < tw(代码14行)，否则h=th-1时，不会抛异常，但20行和21行randint会报错。

如果在调用RandomCrop之前没有pad，此处会有问题。

# torchvision中transforms.py源码截取
class RandomCrop(torch.nn.Module):
    def get_params(img: Tensor, output_size: Tuple[int, int]) -> Tuple[int, int, int, int]:
        """Get parameters for ``crop`` for a random crop.

        Args:
            img (PIL Image or Tensor): Image to be cropped.
            output_size (tuple): Expected output size of the crop.

        Returns:
            tuple: params (i, j, h, w) to be passed to ``crop`` for random crop.
        """
        _, h, w = F.get_dimensions(img)
        th, tw = output_size

        if h + 1 < th or w + 1 < tw:
            raise ValueError(f"Required crop size {(th, tw)} is larger then input image size {(h, w)}")

        if w == tw and h == th:
            return 0, 0, h, w

        i = torch.randint(0, h - th + 1, size=(1,)).item()
        j = torch.randint(0, w - tw + 1, size=(1,)).item()
        return i, j, th, tw

调用测试：

结果：

randint报错。