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

Fixed box.c

Browse Source
This commit is contained in:
Anup Rajput 2017-01-23 11:29:15 +05:30
parent 9d78cf5d00
commit b4b9e8f3c3
1 changed files with 2 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/box.c
View File

@ -249,39 +249,11 @@ int nms_comparator(const void *pa, const void *pb)
void do_nms_obj(box *boxes, float **probs, int total, int classes, float thresh) void do_nms_obj(box *boxes, float **probs, int total, int classes, float thresh)
{ {
int i, j, k; int i, j, k;
sortable_bbox *s = calloc(total, sizeof(sortable_bbox)); sortable_bbox *s = (sortable_bbox *)calloc(total, sizeof(sortable_bbox));
for(i = 0; i < total; ++i){ for(i = 0; i < total; ++i){
s[i].index = i; s[i].index = i;
s[i].class = classes; s[i].class1 = classes;
s[i].probs = probs;
}
qsort(s, total, sizeof(sortable_bbox), nms_comparator);
for(i = 0; i < total; ++i){
if(probs[s[i].index][classes] == 0) continue;
box a = boxes[s[i].index];
for(j = i+1; j < total; ++j){
box b = boxes[s[j].index];
if (box_iou(a, b) > thresh){
for(k = 0; k < classes+1; ++k){
probs[s[j].index][k] = 0;
}
}
}
}
free(s);
}
void do_nms_sort(box *boxes, float **probs, int total, int classes, float thresh)
{
int i, j, k;
sortable_bbox *s = calloc(total, sizeof(sortable_bbox));
for(i = 0; i < total; ++i){
s[i].index = i;
s[i].class = classes;
s[i].probs = probs; s[i].probs = probs;
} }