(image, scale, resample)
| 224 | |
| 225 | |
| 226 | def preprocess(image, scale, resample): |
| 227 | image = image.resize((scale, scale), resample=resample) |
| 228 | image = np.array(image).astype(np.uint8) |
| 229 | image = (image / 127.5 - 1.0).astype(np.float32) |
| 230 | return image |
| 231 | |
| 232 | |
| 233 | def collate_fn(examples, with_prior_preservation): |