Procházet zdrojové kódy

Use gray for label2rgb

Kentaro Wada před 5 roky
rodič
revize
926eca4135
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      labelme/cli/json_to_dataset.py

+ 1 - 1
labelme/cli/json_to_dataset.py

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