def _augment(img, lbl): """An image augmentation function""" img = add_gaussian_noise(img, sigma=0.1) [img, lbl] = flip([img, lbl], axis=1) return img, lbl
def _augment(img): """An image augmentation function""" return add_gaussian_noise(img, sigma=0.1)