PFA-Demo / cvtransforms.py
wissemkarous's picture
init
8c79f36 verified
raw
history blame
273 Bytes
# 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