File size: 257 Bytes
ca51874 |
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
|