New additional classes not training

Question

New additional classes not training

CowbungaMk1 opened this issue a year ago · comments

Hello, I am working on a project that facilitates adding new classes to the cityscapes dataset and adding these classes to dataset/cityscapes.py. However, the training doesn't seem to ever learn these classes and the IOU is always 'nan' for 19-22 (the added classes). If i switch the class ids, such as my new class 22 to 0 (in place of road) it will train that class just fine but not for the other class. Is there some weight initialization step I'm missing? thanks

Gongfan Fang · Answer 1 · Wed Jul 26 2023 12:50:12 GMT+0800 (China Standard Time)

Hello @CowbungaMk1, please modify this line to add new classes.

CowbungaMk1 · Answer 2 · Thu Jul 27 2023 00:31:34 GMT+0800 (China Standard Time)

Thanks for your quick response. I have already tried that and it still trains the original classes just fine. However the additional classes always return a MIOU of 'nan' to visdom and don't appear when using predict.

Thanks

CowbungaMk1 · Answer 3 · Mon Jul 31 2023 23:22:51 GMT+0800 (China Standard Time)

issues resolved