소스 검색

Fix lblsave

Kentaro Wada 5 년 전
부모
커밋
a9f562df67
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      labelme/utils/_io.py

+ 1 - 1
labelme/utils/_io.py

@@ -14,7 +14,7 @@ def lblsave(filename, lbl):
     if lbl.min() >= -1 and lbl.max() < 255:
         lbl_pil = PIL.Image.fromarray(lbl.astype(np.uint8), mode='P')
         colormap = imgviz.label_colormap()
-        lbl_pil.putpalette((colormap * 255).astype(np.uint8).flatten())
+        lbl_pil.putpalette(colormap.flatten())
         lbl_pil.save(filename)
     else:
         raise ValueError(