import numpy as np from PIL import Image g_cn_HWC3 = None def convertIntoCNImageFormat(image): global g_cn_HWC3 if g_cn_HWC3 is None: from annotator.util import HWC3 g_cn_HWC3 = HWC3 color = g_cn_HWC3(np.asarray(image).astype(np.uint8)) return color def convertImageIntoPILFormat(image): return Image.fromarray( np.ascontiguousarray(image.clip(0, 255).astype(np.uint8)).copy() )