yolov5-ncnn-android icon indicating copy to clipboard operation
yolov5-ncnn-android copied to clipboard

confidence计算问题

Open LinzhouLi opened this issue 2 years ago • 1 comments

在文件yolov5_head_ncnn.cpp 中,第167行计算 box_score 的 sigmoid 是不是多余的?因为在下一行代码又对 box_score 做了 sigmoid。这样会导致计算出来的 confidence 偏低。

float box_score = sigmoid(feat_blob.channel(q * feat_offset + 4).row(i)[j]);
float confidence = sigmoid(box_score) * sigmoid(class_score);

LinzhouLi avatar May 15 '22 02:05 LinzhouLi