b072050
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import random def HorizontalFlip(batch_img, p=0.5): # (T, H, W, C) if random.random() > p: batch_img = batch_img[:, :, ::-1, ...] return batch_img def ColorNormalize(batch_img): batch_img = batch_img / 255.0 return batch_img
