search

zhixuhao / unet

unet for image segmentation
MIT License
4.6k stars 2k forks source link

question about segment color #120

Open cena001plus opened 5 years ago

cena001plus commented 5 years ago
  1. I found that CLOOR_DICT is defined, but it is not used in this demo. I think it is necessary to define the color. If you want to use your own defined color on this semantic segmentation image, what should I do?
  2. According to my understanding, the color of the label in the demo image is the final test result of the model. The label is completely black and white (the pixel values ​​are 0 and 255), but if the model is not trained for a long time, The segmentation graph predicted by the model is gray, not completely 0 or 255. What should I do if I want to get the full 0 and 255 predictions?
ledakk commented 5 years ago

Try changing 255 to 255.0

cena001plus commented 5 years ago

Try changing 255 to 255.0

it's not work