mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Added COCO, fixed memory leaks

Browse Source
This commit is contained in:
Joseph Redmon
2015-07-30 16:19:14 -07:00
parent 98813ad520
commit 846b3b4366
10 changed files with 359 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/box.c
View File

@ -211,3 +211,23 @@ dbox diou(box a, box b)
return dd;
}
void do_nms(box *boxes, float **probs, int num_boxes, int classes, float thresh)
{
int i, j, k;
for(i = 0; i < num_boxes*num_boxes; ++i){
int any = 0;
for(k = 0; k < classes; ++k) any = any || (probs[i][k] > 0);
if(!any) {
continue;
}
for(j = i+1; j < num_boxes*num_boxes; ++j){
if (box_iou(boxes[i], boxes[j]) > thresh){
for(k = 0; k < classes; ++k){
if (probs[i][k] < probs[j][k]) probs[i][k] = 0;
else probs[j][k] = 0;
}
}
}
}
}