MrBesen/ffmpeg
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

libavfilter/vf_dnn_detect: bbox index may bigger than bbox number

Browse Source 
Fix a bug that queried bbox index may bigger than bbox's total number.

Signed-off-by: Wenbin Chen <wenbin.chen@intel.com>
This commit is contained in:
Wenbin Chen 2023-07-17 13:33:01 +08:00 committed by Guo Yejun
parent 87f76d3bee
commit 13f49a7c62
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libavfilter/vf_dnn_detect.c
View File

@ -106,12 +106,11 @@ static int dnn_detect_post_proc_ov(AVFrame *frame, DNNData *output, AVFilterCont
float x1 = detections[i * detect_size + 5];
float y1 = detections[i * detect_size + 6];
bbox = av_get_detection_bbox(header, i);
if (conf < conf_threshold) {
continue;
}
bbox = av_get_detection_bbox(header, header->nb_bboxes - nb_bboxes);
bbox->x = (int)(x0 * frame->width);
bbox->w = (int)(x1 * frame->width) - bbox->x;
bbox->y = (int)(y0 * frame->height);