微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

多个带有填充的图像上的相同 RandomCrop 如何获取参数?

如何解决多个带有填充的图像上的相同 RandomCrop 如何获取参数?

我有两个想要以相同方式裁剪的 PIL 图像。它们具有相似的大小,但有时 torchvision.transforms.randomcrop() 会在图像边界之外进行裁剪,从而将生成的张量变为 nan。 为了防止 randomcrop 有一个函数 torchvision.transforms.randomcrop(pad_if_needed=True)

我需要为两个单独的图像使用相同的裁剪参数,所以我需要调用 .get_params 函数

        params = transforms.RandomCrop.get_params(image,output_size=(self.x,self.y))

但我仍然想要 pad_if_needed 功能。当我这样做时:

params = transforms.RandomCrop((self.x,self.y),pad_if_needed=True).get_params(image,self.y))

如果需要,它不使用垫,仍然在图像区域外裁剪。

当我这样做时:

RandCrop = transforms.RandomCrop((self.x,pad_if_needed=True)
params = RandCrop.get_params(image,self.y))

它也不起作用

我怎样才能让它工作?

信息: 我做这样的裁剪:(TF是transforms.functional)

        image = TF.crop(image,params[0],params[1],params[2],params[3])
        mask = TF.crop(mask,params[3])

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。