How can I change codes for submiting label ID png to test?

[zhy168]

I want to use my model to predict test set and save label ID png for submiting result to cityscapes web and get tested.Is there any option to save predict png without using colormap?
thanks a lot !

[zhy168]

If I coment line 171 can work?

[zh320]

Hi, to submit your results to cityscapes please refer to the official documents of cityscapes.

If you want to save predict png without using colormap, you may change the following codes

realtime-semantic-segmentation-pytorch/core/seg_trainer.py

Line 168 in 0e5fe54

preds = self.colormap[preds.max(dim=1)[1]].cpu().numpy()

into

preds = preds.max(dim=1)[1].cpu().numpy()

Best,