rk3399pro运行yolov4，推理结果检测框特别多

使用rk3399pro 运行YOLOv4，推理结果如下图所示，每个物体都有好几个检测框套在一起，大家有碰到类似的问题吗？求大佬赐教！

不做NMS？

jefferyzhang 发表于 2021-6-23 08:59
不做NMS？

做了NMS的，我上按照论坛上面的yolov3 C++代码写的，请问是这里出问题了吗？


float overlap(float x1,float w1,float x2,float w2){
        float l1=x1-w1/2;
        float l2=x2-w2/2;
        float left=l1>l2? l1:l2;
        float r1=x1+w1/2;
        float r2=x2+w2/2;
        float right=r1<r2? r1:r2;
        return right-left;
}

float box_intersection(box a, box b)
{
    float w = overlap(a.x, a.w, b.x, b.w);
    float h = overlap(a.y, a.h, b.y, b.h);
    if(w < 0 || h < 0) return 0;
    float area = w*h;
    return area;
}

float box_union(box a, box b)
{
    float i = box_intersection(a, b);
    float u = a.w*a.h + b.w*b.h - i;
    return u;
}

float box_iou(box a, box b)
{
    return box_intersection(a, b)/box_union(a, b);
}

int nms_comparator(const void *pa, const void *pb)
{
    detection a = *(detection *)pa;
    detection b = *(detection *)pb;
    float diff = 0;
    if(b.sort_class >= 0){
        diff = a.prob[b.sort_class] - b.prob[b.sort_class];
    } else {
        diff = a.objectness - b.objectness;
    }
    if(diff < 0) return 1;
    else if(diff > 0) return -1;
    return 0;
}

int do_nms_sort(detection *dets, int total, int classes, float thresh)
{
    int i, j, k;
    k = total-1;
    for(i = 0; i <= k; ++i){
        if(dets.objectness == 0){
            detection swap = dets;
            dets = dets[k];
            dets[k] = swap;
            --k;
            --i;
        }
    }
    total = k+1;
        //cout<<"total after OBJ_THRESH: "<<total<<"\n";

    for(k = 0; k < classes; ++k){
        for(i = 0; i < total; ++i){
            dets.sort_class = k;
        }
        qsort(dets, total, sizeof(detection), nms_comparator);
        for(i = 0; i < total; ++i){
            if(dets.prob[k] == 0) continue;
            box a = dets.bbox;
            for(j = i+1; j < total; ++j){
                box b = dets[j].bbox;
                if (box_iou(a, b) > thresh){
                    dets[j].prob[k] = 0;
                }
            }
        }
    }
        return total;
}

调整下参数啊...
重叠比例改大不就好了

jefferyzhang 发表于 2021-6-23 09:22
调整下参数啊...
重叠比例改大不就好了

找到问题了，outputs_transform函数里面的mask_0,mask_1,mask_2和anchors的值 需要按照yolov4.cfg里面的设置做出相应的修改，
谢谢大佬的提醒