highperfocused/ffmpeg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

libavfilter/vf_dnn_detect: Use class confidence to filt boxes

Browse Source 
Use class confidence instead of box_score to filt boxes, which is more
accurate. Class confidence is obtained by multiplying class probability
distribution and box_score.

Signed-off-by: Wenbin Chen <wenbin.chen@intel.com>
Reviewed-by: Guo Yejun <yejun.guo@intel.com>
This commit is contained in:
Wenbin Chen 2024-01-17 15:21:51 +08:00 committed by Guo Yejun
parent 3de38b9da5
commit d52e8ed114
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libavfilter/vf_dnn_detect.c
View File

@ -236,9 +236,6 @@ static int dnn_detect_parse_yolo_output(AVFrame *frame, DNNData *output, int out
conf = post_process_raw_data(
detection_boxes_data[cy * cell_w + cx + 4 * cell_w * cell_h]);
}
if (conf < conf_threshold) {
continue;
}
if (is_NHWC) {
x = post_process_raw_data(detection_boxes_data[0]);
@ -257,6 +254,9 @@ static int dnn_detect_parse_yolo_output(AVFrame *frame, DNNData *output, int out
conf = conf * post_process_raw_data(
detection_boxes_data[cy * cell_w + cx + (label_id + 5) * cell_w * cell_h]);
}
if (conf < conf_threshold) {
continue;
}
bbox = av_mallocz(sizeof(*bbox));
if (!bbox)