(train_x, val_x)
| 69 | |
| 70 | |
| 71 | def normalize(train_x, val_x): |
| 72 | mean = [0.4914, 0.4822, 0.4465] |
| 73 | std = [0.2023, 0.1994, 0.2010] |
| 74 | train_x /= 255 |
| 75 | val_x /= 255 |
| 76 | for ch in range(0, 2): |
| 77 | train_x[:, ch, :, :] -= mean[ch] |
| 78 | train_x[:, ch, :, :] /= std[ch] |
| 79 | val_x[:, ch, :, :] -= mean[ch] |
| 80 | val_x[:, ch, :, :] /= std[ch] |
| 81 | return train_x, val_x |
| 82 | |
| 83 | def load(): |
| 84 | train_x, train_y = load_train_data() |