ShawDa/unet-rgb

about save_img in unet_camvid.py

Opened this issue · 0 comments

Scikit-image (skimage) has a method called color.label2rgb which converts to an array of labels to a colored image. So you don't have to implement the coloring algorithm yourself.