Переглянути джерело

Replace labelme.cli.draw_label_png with imgviz

Jongha Lim 5 роки тому
батько
коміт
68fb3ebc4b
1 змінених файлів з 2 додано та 2 видалено
  1. 2 2
      labelme/cli/draw_label_png.py

+ 2 - 2
labelme/cli/draw_label_png.py

@@ -1,11 +1,11 @@
 import argparse
 
+import imgviz
 import matplotlib.pyplot as plt
 import numpy as np
 import PIL.Image
 
 from labelme.logger import logger
-from labelme import utils
 
 
 def main():
@@ -19,7 +19,7 @@ def main():
     logger.info('label shape: {}'.format(lbl.shape))
     logger.info('unique label values: {}'.format(np.unique(lbl)))
 
-    lbl_viz = utils.draw_label(lbl)
+    lbl_viz = imgviz.label2rgb(lbl)
     plt.imshow(lbl_viz)
     plt.show()