mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-27 10:44:19 +00:00
15 lines
273 B
Python
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
|