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