| 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() | |
| ) | |