def TF_enhance_brightness(x, p=1.0): assert len(x.shape) == 3 h, w, nc = x.shape enhancer = ImageEnhance.Brightness(np_to_pil(x)) return pil_to_np(enhancer.enhance(p))