'log.json' file contains incorrect bounding boxes

sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset

MIT License

223 stars 77 forks source link

'log.json' file contains incorrect bounding boxes #135

Closed jgmv closed 3 months ago

jgmv commented 7 months ago

The 'log.json' file generated when using the option 'inference.py --log-json' contains more bounding boxes than those detected in the images. This often includes objects from classes not displayed in the images.

sovit-123 commented 7 months ago

Will take a look into it ASAP. Need to check the code. Maybe a threshold issue.

jgmv commented 7 months ago

Yes, I think the problem is that the boxes are saved before the threshold is applied

joshua-atolagbe commented 3 months ago

@sovit-123 Is this solved?

sovit-123 commented 3 months ago

@joshua-atolagbe @jgmv I just pushed an update. It's corrected now. Please pull.