importos.pathasospimportnumpyasnpimportPIL.Imagefromlabelme.utils.drawimportlabel_colormapdeflblsave(filename,lbl):ifosp.splitext(filename)[1]!='.png':filename+='.png'# Assume label ranses [-1, 254] for int32,# and [0, 255] for uint8 as VOC.iflbl.min()>=-1andlbl.max()<255:lbl_pil=PIL.Image.fromarray(lbl.astype(np.uint8),mode='P')colormap=label_colormap(255)lbl_pil.putpalette((colormap*255).astype(np.uint8).flatten())lbl_pil.save(filename)else: