inference label is wrong

[Amore-HDU]

Hello， I have trained FCOS on my own dataset. But when I tried to test my model, I found bounding box was right while label was wrong. I have changed CATEGORIES in demo/predictor.py. @tianzhi0549

[Amore-HDU]

I found when I trained my FCOS, it occured masks = [obj["segmentation"] for obj in anno] KeyError: 'segmentation' so I comment masks = [obj["segmentation"] for obj in anno] masks = SegmentationMask(masks, img.size, mode='poly') target.add_field("masks", masks). I think the reason of wrong label can be it

[Mahmood-Hussain]

I found when I trained my FCOS, it occured masks = [obj["segmentation"] for obj in anno] KeyError: 'segmentation' so I comment masks = [obj["segmentation"] for obj in anno] masks = SegmentationMask(masks, img.size, mode='poly') target.add_field("masks", masks). I think the reason of wrong label can be it

hello did you find what was exactly wrong with your code ? I am also getting masks = [obj["segmentation"] for obj in anno] KeyError: 'segmentation'

[Mahmood-Hussain]

Hello， I have trained FCOS on my own dataset. But when I tried to test my model, I found bounding box was right while label was wrong. I have changed CATEGORIES in demo/predictor.py. @tianzhi0549

My labels are also wrong. Did you get an answer for this? @Amore-HDU