(x, mean=[0.485, 0.456, 0.406], std=[0.228, 0.224, 0.225])
| 242 | |
| 243 | |
| 244 | def color_normalize(x, mean=[0.485, 0.456, 0.406], std=[0.228, 0.224, 0.225]): |
| 245 | for t, m, s in zip(x, mean, std): |
| 246 | t.sub_(m) |
| 247 | t.div_(s) |
| 248 | return x |
| 249 | |
| 250 | |
| 251 | if __name__ == '__main__': |