如何解决多个带有填充的图像上的相同 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 举报,一经查实,本站将立刻删除。