Pytorch-CustomDataset-FasterRCNN
Pytorch-CustomDataset-FasterRCNN copied to clipboard
Each object is bound by multiple bounding boxes in result
I have trained the model with my dataset. I see in the results each object is identified multiple times. How do I avoid find the same object more than once?
Its been a while since I ran this code: but try changing the threshold values in the calc_iou function