ifnspaml / SGDepth

[ECCV 2020] Self-Supervised Monocular Depth Estimation: Solving the Dynamic Object Problem by Semantic Guidance
MIT License
200 stars 26 forks source link

Modify the cityscapes semantic labels #35

Closed Luara1 closed 1 year ago

Luara1 commented 1 year ago

Hello, I'm sorry to bother you. I have encountered some problems in using this network. When segment, I want to use the categoryID of cityscapes for training. I put the cityscapes in the colors folder TRID in py TO LABEL = labels cityscape seg. Gettrainid2label() is modified to TRID TO LABEL = labels cityscape seg.getcategory2label()。 However, the trained model is still 19 classes, not 7. I want to know how to correctly modify the training semantic labels. I hope to get your answer. Thank you very muchs