AlexeyAB
2018-03-20 fe4f01b1d8cce9762e47aaf42b6e17a3427a144b
src/box.c
@@ -232,7 +232,7 @@
typedef struct{
    int index;
    int class;
    int class_id;
    float **probs;
} sortable_bbox;
@@ -240,7 +240,7 @@
{
    sortable_bbox a = *(sortable_bbox *)pa;
    sortable_bbox b = *(sortable_bbox *)pb;
    float diff = a.probs[a.index][b.class] - b.probs[b.index][b.class];
    float diff = a.probs[a.index][b.class_id] - b.probs[b.index][b.class_id];
    if(diff < 0) return 1;
    else if(diff > 0) return -1;
    return 0;
@@ -253,13 +253,13 @@
    for(i = 0; i < total; ++i){
        s[i].index = i;       
        s[i].class = 0;
        s[i].class_id = 0;
        s[i].probs = probs;
    }
    for(k = 0; k < classes; ++k){
        for(i = 0; i < total; ++i){
            s[i].class = k;
            s[i].class_id = k;
        }
        qsort(s, total, sizeof(sortable_bbox), nms_comparator);
        for(i = 0; i < total; ++i){