search

mhamilton723 / STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences
MIT License
712 stars 143 forks source link

Gray images results #14

Closed alien19 closed 2 years ago

alien19 commented 2 years ago

Hello @mhamilton723, thanks for your great efforts. I have tried the demo_segmeentation on custom images with the provided models but the resulted images in the cluster folder are just in black and grey as shown. ship_containers_human What might be the reason for that? or how can I visualize the results in colors as shown in your work?

Thanks in advance!

mhamilton723 commented 2 years ago

Hey @alien19 this is as intended because the are PNG's with pixel values in the range of [0.....NumClasses]. To visualize them with a different colormap, you can load the images in python, and convert class numbers to colors of your choice.