icefall/egs/grid/AVSR/local/cvtransforms.py

15 lines
273 B
Python

# 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