浏览代码

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