| # 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 | |