浏览代码

Use imgviz.asgray

Close https://github.com/wkentaro/labelme/issues/588
Kentaro Wada 5 年之前
父节点
当前提交
c6266dcf9e
共有 2 个文件被更改,包括 3 次插入2 次删除
  1. 2 1
      labelme/cli/json_to_dataset.py
  2. 1 1
      setup.py

+ 2 - 1
labelme/cli/json_to_dataset.py

@@ -56,8 +56,9 @@ def main():
     label_names = [None] * (max(label_name_to_value.values()) + 1)
     for name, value in label_name_to_value.items():
         label_names[value] = name
+
     lbl_viz = imgviz.label2rgb(
-        label=lbl, img=imgviz.rgb2gray(img), label_names=label_names, loc='rb'
+        label=lbl, img=imgviz.asgray(img), label_names=label_names, loc='rb'
     )
 
     PIL.Image.fromarray(img).save(osp.join(out_dir, 'img.png'))

+ 1 - 1
setup.py

@@ -27,7 +27,7 @@ def get_install_requires():
     assert PY3 or PY2
 
     install_requires = [
-        'imgviz>=0.10.6',
+        'imgviz>=0.11.0',
         'matplotlib',
         'numpy',
         'Pillow>=2.8.0',