(image, target, padding)
| 152 | |
| 153 | |
| 154 | def pad(image, target, padding): |
| 155 | # assumes that we only pad on the bottom right corners |
| 156 | padded_image = F.pad(image, (0, 0, padding[0], padding[1])) |
| 157 | if target is None: |
| 158 | return padded_image, None |
| 159 | target = target.copy() |
| 160 | # should we do something wrt the original size? |
| 161 | target["size"] = torch.tensor(padded_image.size[::-1]) |
| 162 | if "masks" in target: |
| 163 | target['masks'] = torch.nn.functional.pad(target['masks'], (0, padding[0], 0, padding[1])) |
| 164 | return padded_image, target |
| 165 | |
| 166 | |
| 167 | class RandomCrop(object): |
no test coverage detected