|
@@ -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'))
|