yolov5-ncnn-android
yolov5-ncnn-android copied to clipboard
confidence计算问题
在文件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);