dnn_filter_common.h File Reference

#include "dnn_interface.h"

Go to the source code of this file.

Data Structures
struct	DnnContext

Macros
#define	DNN_COMMON_OPTIONS

Functions
int	ff_dnn_init (DnnContext *ctx, DNNFunctionType func_type, AVFilterContext *filter_ctx)
int	ff_dnn_set_frame_proc (DnnContext *ctx, FramePrePostProc pre_proc, FramePrePostProc post_proc)
int	ff_dnn_set_detect_post_proc (DnnContext *ctx, DetectPostProc post_proc)
int	ff_dnn_set_classify_post_proc (DnnContext *ctx, ClassifyPostProc post_proc)
int	ff_dnn_get_input (DnnContext *ctx, DNNData *input)
int	ff_dnn_get_output (DnnContext *ctx, int input_width, int input_height, int *output_width, int *output_height)
int	ff_dnn_execute_model (DnnContext *ctx, AVFrame *in_frame, AVFrame *out_frame)
int	ff_dnn_execute_model_classification (DnnContext *ctx, AVFrame *in_frame, AVFrame *out_frame, const char *target)
DNNAsyncStatusType	ff_dnn_get_result (DnnContext *ctx, AVFrame **in_frame, AVFrame **out_frame)
int	ff_dnn_flush (DnnContext *ctx)
void	ff_dnn_uninit (DnnContext *ctx)

Detailed Description

common functions for the dnn based filters

Definition in file dnn_filter_common.h.

Macro Definition Documentation

DNN_COMMON_OPTIONS

#define DNN_COMMON_OPTIONS

Value:

    { "model",              "path to model file",         OFFSET(model_filename),   AV_OPT_TYPE_STRING,    { .str = NULL }, 0, 0, FLAGS },\
    { "input",              "input name of the model",    OFFSET(model_inputname),  AV_OPT_TYPE_STRING,    { .str = NULL }, 0, 0, FLAGS },\
    { "output",             "output name of the model",   OFFSET(model_outputnames_string), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },\
    { "backend_configs",    "backend configs",            OFFSET(backend_options),  AV_OPT_TYPE_STRING,    { .str = NULL }, 0, 0, FLAGS },\
    { "options", "backend configs (deprecated, use backend_configs)", OFFSET(backend_options),  AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS | AV_OPT_FLAG_DEPRECATED},\
    { "async",              "use DNN async inference (ignored, use backend_configs='async=1')",    OFFSET(async),            AV_OPT_TYPE_BOOL,      { .i64 = 1},     0, 1, FLAGS},

Definition at line 43 of file dnn_filter_common.h.

Function Documentation

ff_dnn_init()

int ff_dnn_init	(	DnnContext *	ctx,
		DNNFunctionType	func_type,
		AVFilterContext *	filter_ctx
	)

Definition at line 54 of file dnn_filter_common.c.

Referenced by dnn_classify_init(), dnn_detect_init(), and init().

ff_dnn_set_frame_proc()

int ff_dnn_set_frame_proc	(	DnnContext *	ctx,
		FramePrePostProc	pre_proc,
		FramePrePostProc	post_proc
	)

Definition at line 90 of file dnn_filter_common.c.

ff_dnn_set_detect_post_proc()

int ff_dnn_set_detect_post_proc	(	DnnContext *	ctx,
		DetectPostProc	post_proc
	)

Definition at line 97 of file dnn_filter_common.c.

Referenced by dnn_detect_init().

ff_dnn_set_classify_post_proc()

int ff_dnn_set_classify_post_proc	(	DnnContext *	ctx,
		ClassifyPostProc	post_proc
	)

Definition at line 103 of file dnn_filter_common.c.

Referenced by dnn_classify_init().

ff_dnn_get_input()

int ff_dnn_get_input	(	DnnContext *	ctx,
		DNNData *	input
	)

Definition at line 109 of file dnn_filter_common.c.

Referenced by config_input().

ff_dnn_get_output()

int ff_dnn_get_output	(	DnnContext *	ctx,
		int	input_width,
		int	input_height,
		int *	output_width,
		int *	output_height
	)

Definition at line 114 of file dnn_filter_common.c.

Referenced by config_output().

ff_dnn_execute_model()

int ff_dnn_execute_model	(	DnnContext *	ctx,
		AVFrame *	in_frame,
		AVFrame *	out_frame
	)

Definition at line 120 of file dnn_filter_common.c.

Referenced by activate(), dnn_detect_activate(), and filter_frame().

ff_dnn_execute_model_classification()

int ff_dnn_execute_model_classification	(	DnnContext *	ctx,
		AVFrame *	in_frame,
		AVFrame *	out_frame,
		const char *	target
	)

Definition at line 132 of file dnn_filter_common.c.

Referenced by dnn_classify_activate().

ff_dnn_get_result()

DNNAsyncStatusType ff_dnn_get_result	(	DnnContext *	ctx,
		AVFrame **	in_frame,
		AVFrame **	out_frame
	)

Definition at line 147 of file dnn_filter_common.c.

Referenced by activate(), dnn_classify_activate(), dnn_classify_flush_frame(), dnn_detect_activate(), dnn_detect_flush_frame(), filter_frame(), and flush_frame().

ff_dnn_flush()

int ff_dnn_flush

(

DnnContext *

ctx

)

Definition at line 152 of file dnn_filter_common.c.

Referenced by dnn_classify_flush_frame(), dnn_detect_flush_frame(), and flush_frame().

ff_dnn_uninit()

void ff_dnn_uninit

(

DnnContext *

ctx

)

Definition at line 157 of file dnn_filter_common.c.

Referenced by dnn_classify_uninit(), dnn_detect_uninit(), and uninit().