LIPREAD / cvtransforms.py
wissemkarous's picture
init
b072050 verified
raw
history blame
257 Bytes
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