(self, images)
| 31 | return image |
| 32 | |
| 33 | def process_images(self, images): |
| 34 | images = [self.process_image(image) for image in images] |
| 35 | images = torch.stack(images) |
| 36 | return images |
| 37 | |
| 38 | def decode_images(self, images): |
| 39 | images = (images[:, [2,1,0]].permute(0, 2, 3, 1) * 255).clip(0, 255).numpy().astype(np.uint8) |
no test coverage detected