search

ayooshkathuria / YOLO_v3_tutorial_from_scratch

Accompanying code for Paperspace tutorial series "How to Implement YOLO v3 Object Detector from Scratch"
https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/
2.32k stars 724 forks source link

About NMS #50

Open bot66 opened 5 years ago

bot66 commented 5 years ago

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...