# coding: utf-8 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