(x, image_size)
| 257 | |
| 258 | |
| 259 | def resize_dataset(x, image_size): |
| 260 | num_data = x.shape[0] |
| 261 | dim = x.shape[1] |
| 262 | X = np.zeros(shape=(num_data, dim, image_size, image_size), |
| 263 | dtype=np.float32) |
| 264 | for n in range(0, num_data): |
| 265 | for d in range(0, dim): |
| 266 | X[n, d, :, :] = np.array(Image.fromarray(x[n, d, :, :]).resize( |
| 267 | (image_size, image_size), Image.BILINEAR), |
| 268 | dtype=np.float32) |
| 269 | return X |
| 270 | |
| 271 | |
| 272 | def run(global_rank, |