def _augment(img):
    if hflip:
        img = np.flip(img, axis=1)
    if vflip:
        img = np.flip(img, axis=0)
    if rot90:
        img = np.rot90(img, 1)
    return img
def _augment(img):
    if hflip:
        img = np.flip(img, axis=1)
    if vflip:
        img = np.flip(img, axis=0)
    if rot90:
        img = np.rot90(img, 1)
    return img