# coding: utf-8 import random import cv2 import numpy as np 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