warmspringwinds / tf-image-segmentation

Image Segmentation framework based on Tensorflow and TF-Slim library
MIT License
549 stars 188 forks source link

colored segmentation to gray class value png #18

Closed ahundt closed 7 years ago

ahundt commented 7 years ago

You mentioned there is a more efficient function for converting the colored segmentation pngs to gray class pngs where the value is the class number. Which is the preferred function to do this?

warmspringwinds commented 7 years ago

We read the png file in the following way: https://github.com/warmspringwinds/tf-image-segmentation/blob/master/tf_image_segmentation/utils/tf_records.py#L37

This give you the classes and not the rgb values

ahundt commented 7 years ago

Thanks!