I am trying to train the RTMDet-X model using my custom dataset.
I use the coco metric to evaluate the model and get a bbox of over 70 %, while the segm AP stays at 30-40 %.
The masks are correct as other models achieve AP higher that 65 %.
I am training on a single GPU.
My guess that something in the config is not correct.
Checklist
Describe the bug
I am trying to train the RTMDet-X model using my custom dataset. I use the coco metric to evaluate the model and get a bbox of over 70 %, while the segm AP stays at 30-40 %. The masks are correct as other models achieve AP higher that 65 %. I am training on a single GPU. My guess that something in the config is not correct.
Reproduction
See config below.
My own dataset, size of train is about 1200 images, with 300 test and 21 classes.
Environment
Environment installed inside of a conda evnrionment.
Installed via this commands:
Config If applicable, paste the error trackback here.
Final training output exmaple