(self, image_batch, padding_factor)
| 222 | return image_batch |
| 223 | |
| 224 | def expandImagePad(self, image_batch, padding_factor): |
| 225 | b, c, h, w = image_batch.size() |
| 226 | expand_image = torch.zeros([b, c, h * 2, w * 2]) |
| 227 | expand_image[:, :, h//2: h//2 + h, w//2: w//2 + w] = image_batch |
| 228 | |
| 229 | return expand_image |
| 230 | |
| 231 | |
| 232 |