smhassanerfani/atlantis

Segmentation mask image is three color channel

prajachintya opened this issue · 8 comments

Hi,

I've tried to export images to Segmentation Mask in CVAT, but it came out as three color channel image. How can I get the single-channel mask images?

Thank you :)

Hi,
Unfortunately, CVAT doesn't give you ID maps. I suggest using color2id.py. Just config the labels_info.json based on your own labels list.

Good luck!

Thank you for the reply. This algorithm works fine. Btw how did you do the evaluation for the prediction results?

Hi, Sorry I was a little busy this week, I'll add argparse package to color2id.py to be easier to work with it. Regarding your second question, you can use compute_iou.py. Again, please configure the labels_info.json for this code.

Good Luck!

Thank you!

According to the manuscript of the paper, "The aquatic path is to segment different types of waterbodies including sea, river, waterfall, wetland and etc., while the non-aquatic path is to segment other categories such as ship and bridge." The manuscript explains further that "Aquatic categories include 17 labels showing just water content in different forms and bodies, e.g., sea, river, lake, etc."

Hope it helps.
Mohammad

Yes! AQUANet is customized for ATLANTIS. I would modify the self.indices based on the new dataset.