File size: 433 Bytes
194b4ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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()
)
|