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

avfilter/find_rect: add option to discard non-matching frames

Browse Source 
Default is disabled.
This commit is contained in:
Gyan Doshi 2021-04-01 18:22:03 +05:30
parent aff23c3474
commit abdafca9ad
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
doc/filters.texi
View File

@ -12139,6 +12139,9 @@ Number of mipmaps, default is 3.
@item xmin, ymin, xmax, ymax
Specifies the rectangle in which to search.
@item discard
Discard frames where object is not detected. Default is disabled.
@end table
@subsection Examples

9
libavfilter/vf_find_rect.c
View File

@ -40,6 +40,7 @@ typedef struct FOCContext {
AVFrame *obj_frame;
AVFrame *needle_frame[MAX_MIPMAPS];
AVFrame *haystack_frame[MAX_MIPMAPS];
int discard;
} FOCContext;
#define OFFSET(x) offsetof(FOCContext, x)
@ -52,6 +53,7 @@ static const AVOption find_rect_options[] = {
{ "ymin", "", OFFSET(ymin), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, FLAGS },
{ "xmax", "", OFFSET(xmax), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, FLAGS },
{ "ymax", "", OFFSET(ymax), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, FLAGS },
{ "discard", "", OFFSET(discard), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, FLAGS },
{ NULL }
};
@ -206,7 +208,12 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
}
if (best_score > foc->threshold) {
return ff_filter_frame(ctx->outputs[0], in);
if (foc->discard) {
av_frame_free(&in);
return 0;
} else {
return ff_filter_frame(ctx->outputs[0], in);
}
}
av_log(ctx, AV_LOG_INFO, "Found at n=%lld pts_time=%f x=%d y=%d with score=%f\n",