EfficientDet.Pytorch icon indicating copy to clipboard operation
EfficientDet.Pytorch copied to clipboard

Why are there two thresholds?

Open ma3252788 opened this issue 4 years ago • 3 comments

Can you say what these two thresholds represent?

threshold and iou_threshold

Thank you!!!!

ma3252788 avatar Jan 09 '20 16:01 ma3252788

one(former) is to filter some anchors for those cls-score less than threshold , anthor(latter) is to reduce some anchors for overlap

y78h11b09 avatar Jan 11 '20 01:01 y78h11b09

one(former) is to filter some anchors for those cls-score less than threshold , anthor(latter) is to reduce some anchors for overlap

Thanks for the reply, why does the detection result not change after I adjust these two thresholds? I set both to 0.5, but there are still a lot of boxes.

ma3252788 avatar Jan 12 '20 07:01 ma3252788

one(former) is to filter some anchors for those cls-score less than threshold , anthor(latter) is to reduce some anchors for overlap

Thanks for the reply, why does the detection result not change after I adjust these two thresholds? I set both to 0.5, but there are still a lot of boxes.

That's because although demo.py takes these arguments, no further filtering is then performed using them. After appending boxes, labels and scores, add this:

dataframe = list(zip(bboxes, labels, bbox_scores))
dataframe = list(filter(lambda x: x[2] > args.threshold*100, dataframe))
bboxes, labels, bbox_scores = list(zip(*dataframe))

Best wishes.

xafarranxera avatar Jan 23 '20 16:01 xafarranxera