def TF_enhance_contrast(x, p=1.0, target=None):
assert len(x.shape) == 3
h, w, nc = x.shape
enhancer = ImageEnhance.Contrast(np_to_pil(x))
return pil_to_np(enhancer.enhance(p))
微信
公众号
扫码关注公众号