(image, name)
| 179 | |
| 180 | |
| 181 | def save_image(image, name): |
| 182 | import numpy as np |
| 183 | from PIL import Image |
| 184 | |
| 185 | image = image[0].cpu().numpy() |
| 186 | image = (image + 1.0) * 127.5 |
| 187 | image = image.clip(0, 255).astype(np.uint8) |
| 188 | image = Image.fromarray(image.transpose(1, 2, 0)) |
| 189 | image.save(name) |
| 190 | |
| 191 | |
| 192 | def load_image(uri, size=None, center_crop=False): |
no test coverage detected