#include "dnn_filter_common.h"
#include "libavutil/avstring.h"
#include "libavutil/mem.h"

Macros
#define	MAX_SUPPORTED_OUTPUTS_NB 4

Functions
static char **	separate_output_names (const char expr, const char val_sep, int *separated_nb)

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)

Macro Definition Documentation

◆ MAX_SUPPORTED_OUTPUTS_NB

#define MAX_SUPPORTED_OUTPUTS_NB 4

Definition at line 23 of file dnn_filter_common.c.

Function Documentation

◆ separate_output_names()

static char** separate_output_names	(	const char *	expr,
		const char *	val_sep,
		int *	separated_nb
	)

static

Definition at line 25 of file dnn_filter_common.c.

Referenced by ff_dnn_init().

◆ ff_dnn_init()

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

Definition at line 55 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 103 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 110 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 116 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 122 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 127 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 135 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 147 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 162 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 167 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 172 of file dnn_filter_common.c.

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

Macros

Functions