YOLO_v3_tutorial_from_scratch icon indicating copy to clipboard operation
YOLO_v3_tutorial_from_scratch copied to clipboard

Published 20 hours ago • verified publisher icon ayooshkathuria

About NMS

Open bot66 opened this issue 6 years ago • 0 comments

Why sort bbox with objectness score (Pobj) rather than individual class score (Pc) util.py line 149 conf_sort_index = torch.sort(image_pred_class[:,4], descending = True )[1] I searched NMS sort bbox with Pc and compute the rest bbox IoU with Top Pc score bbox then throw away bboxes IoU bigger than treshhold. And this code is Pobj...

bot66 avatar Sep 28 '19 12:09 bot66