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

avfilter/find_rect: write score to metadata

Browse Source
This commit is contained in:
Gyan Doshi
2021-04-01 18:46:20 +05:30
parent abdafca9ad
commit 18dcbb0d6c
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libavfilter/vf_find_rect.c
View File

@ -187,6 +187,7 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
float best_score; float best_score;
int best_x, best_y; int best_x, best_y;
int i; int i;
char buf[32];
foc->haystack_frame[0] = av_frame_clone(in); foc->haystack_frame[0] = av_frame_clone(in);
for (i=1; i<foc->mipmaps; i++) { for (i=1; i<foc->mipmaps; i++) {
@ -222,12 +223,15 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
foc->last_x = best_x; foc->last_x = best_x;
foc->last_y = best_y; foc->last_y = best_y;
snprintf(buf, sizeof(buf), "%f", best_score);
av_frame_make_writable(in); av_frame_make_writable(in);
av_dict_set_int(&in->metadata, "lavfi.rect.w", foc->obj_frame->width, 0); av_dict_set_int(&in->metadata, "lavfi.rect.w", foc->obj_frame->width, 0);
av_dict_set_int(&in->metadata, "lavfi.rect.h", foc->obj_frame->height, 0); av_dict_set_int(&in->metadata, "lavfi.rect.h", foc->obj_frame->height, 0);
av_dict_set_int(&in->metadata, "lavfi.rect.x", best_x, 0); av_dict_set_int(&in->metadata, "lavfi.rect.x", best_x, 0);
av_dict_set_int(&in->metadata, "lavfi.rect.y", best_y, 0); av_dict_set_int(&in->metadata, "lavfi.rect.y", best_y, 0);
av_dict_set(&in->metadata, "lavfi.rect.score", buf, 0);
return ff_filter_frame(ctx->outputs[0], in); return ff_filter_frame(ctx->outputs[0], in);
} }