search

hellochick / ICNet-tensorflow

TensorFlow-based implementation of "ICNet for Real-Time Semantic Segmentation on High-Resolution Images".
405 stars 153 forks source link

channel problem of grayscale image #87

Open dbredep opened 5 years ago

dbredep commented 5 years ago

I have encountered the channel problem when i try to use the grayscale image for training. How to solve the three channel problem but can still use the iterator?

hellochick commented 5 years ago

Hey @dbredep,

You can take a look at the parsing function in utils.image_reader.py. What you need to do is to change the decode function!