dnn_interface.h File Reference

DNN inference engine interface. More...

#include <stdint.h>
#include "libavutil/frame.h"
#include "avfilter.h"

Go to the source code of this file.

Data Structures
struct	DNNData
struct	DNNExecBaseParams
struct	DNNExecClassificationParams
struct	DNNModel
struct	TFOptions
struct	OVOptions
struct	THOptions
struct	DnnContext
struct	DNNModule

Macros
#define	DNN_GENERIC_ERROR   FFERRTAG('D','N','N','!')

Typedefs
typedef int(*	FramePrePostProc) (AVFrame *frame, DNNData *model, AVFilterContext *filter_ctx)
typedef int(*	DetectPostProc) (AVFrame *frame, DNNData *output, uint32_t nb, AVFilterContext *filter_ctx)
typedef int(*	ClassifyPostProc) (AVFrame *frame, DNNData *output, uint32_t bbox_index, AVFilterContext *filter_ctx)

Enumerations
enum	DNNBackendType { DNN_TF = 1 , DNN_OV = 1 << 1 , DNN_TH = 1 << 2 , DNN_ONNX = 1 << 3 }
enum	DNNDataType { DNN_FLOAT = 1 , DNN_UINT8 = 4 }
enum	DNNColorOrder { DCO_NONE , DCO_BGR , DCO_RGB }
enum	DNNAsyncStatusType { DAST_FAIL , DAST_EMPTY_QUEUE , DAST_NOT_READY , DAST_SUCCESS }
enum	DNNFunctionType { DFT_NONE , DFT_PROCESS_FRAME , DFT_ANALYTICS_DETECT , DFT_ANALYTICS_CLASSIFY }
enum	DNNLayout { DL_NONE , DL_NCHW , DL_NHWC }

Functions
const DNNModule *	ff_get_dnn_module (DNNBackendType backend_type, void *log_ctx)
void	ff_dnn_init_child_class (DnnContext *ctx)
void *	ff_dnn_child_next (DnnContext *obj, void *prev)
const AVClass *	ff_dnn_child_class_iterate_with_mask (void **iter, uint32_t backend_mask)
static int	dnn_get_width_idx_by_layout (DNNLayout layout)
static int	dnn_get_height_idx_by_layout (DNNLayout layout)
static int	dnn_get_channel_idx_by_layout (DNNLayout layout)

Detailed Description

DNN inference engine interface.

Definition in file dnn_interface.h.

Macro Definition Documentation

DNN_GENERIC_ERROR

#define DNN_GENERIC_ERROR   FFERRTAG('D','N','N','!')

Definition at line 33 of file dnn_interface.h.

Referenced by dnn_flush_ov(), execute_model_onnx(), execute_model_ov(), execute_model_tf(), execute_model_th(), ff_dnn_async_module_cleanup(), ff_dnn_start_inference_async(), fill_model_input_ov(), fill_model_input_tf(), get_input_ov(), get_input_tf(), get_output_ov(), init_model_ov(), load_tf_model(), onnx_start_inference(), tf_start_inference(), and th_start_inference().

Typedef Documentation

FramePrePostProc

typedef int(* FramePrePostProc) (AVFrame *frame, DNNData *model, AVFilterContext *filter_ctx)

Definition at line 94 of file dnn_interface.h.

DetectPostProc

typedef int(* DetectPostProc) (AVFrame *frame, DNNData *output, uint32_t nb, AVFilterContext *filter_ctx)

Definition at line 95 of file dnn_interface.h.

ClassifyPostProc

typedef int(* ClassifyPostProc) (AVFrame *frame, DNNData *output, uint32_t bbox_index, AVFilterContext *filter_ctx)

Definition at line 96 of file dnn_interface.h.

Enumeration Type Documentation

DNNBackendType

enum DNNBackendType

Enumerator
DNN_TF
DNN_OV
DNN_TH
DNN_ONNX

Definition at line 35 of file dnn_interface.h.

DNNDataType

enum DNNDataType

Enumerator
DNN_FLOAT
DNN_UINT8

Definition at line 42 of file dnn_interface.h.

DNNColorOrder

enum DNNColorOrder

Enumerator
DCO_NONE
DCO_BGR
DCO_RGB

Definition at line 44 of file dnn_interface.h.

DNNAsyncStatusType

enum DNNAsyncStatusType

Enumerator
DAST_FAIL
DAST_EMPTY_QUEUE
DAST_NOT_READY
DAST_SUCCESS

Definition at line 50 of file dnn_interface.h.

DNNFunctionType

enum DNNFunctionType

Enumerator
DFT_NONE
DFT_PROCESS_FRAME
DFT_ANALYTICS_DETECT
DFT_ANALYTICS_CLASSIFY

Definition at line 57 of file dnn_interface.h.

DNNLayout

enum DNNLayout

Enumerator
DL_NONE
DL_NCHW
DL_NHWC

Definition at line 64 of file dnn_interface.h.

Function Documentation

ff_get_dnn_module()

const DNNModule * ff_get_dnn_module	(	DNNBackendType	backend_type,
		void *	log_ctx )

Definition at line 93 of file dnn_interface.c.

Referenced by ff_dnn_init().

ff_dnn_init_child_class()

void ff_dnn_init_child_class

(

DnnContext *

ctx

)

Definition at line 106 of file dnn_interface.c.

Referenced by ff_dnn_filter_init_child_class().

ff_dnn_child_next()

void * ff_dnn_child_next	(	DnnContext *	obj,
		void *	prev )

Definition at line 116 of file dnn_interface.c.

Referenced by ff_dnn_filter_child_next(), and ff_dnn_init().

ff_dnn_child_class_iterate_with_mask()

const AVClass * ff_dnn_child_class_iterate_with_mask	(	void **	iter,
		uint32_t	backend_mask )

Definition at line 136 of file dnn_interface.c.

dnn_get_width_idx_by_layout()

static int dnn_get_width_idx_by_layout ( DNNLayout layout )

inlinestatic

Definition at line 209 of file dnn_interface.h.

Referenced by allocate_input_tensor(), check_modelinput_inlink(), config_input(), ff_frame_to_dnn_classify(), ff_frame_to_dnn_detect(), fill_model_input_onnx(), fill_model_input_th(), get_input_ov(), infer_completion_callback(), infer_completion_callback(), infer_completion_callback(), and infer_completion_callback().

dnn_get_height_idx_by_layout()

static int dnn_get_height_idx_by_layout ( DNNLayout layout )

inlinestatic

Definition at line 214 of file dnn_interface.h.

Referenced by allocate_input_tensor(), check_modelinput_inlink(), config_input(), ff_frame_to_dnn_classify(), ff_frame_to_dnn_detect(), fill_model_input_onnx(), fill_model_input_th(), get_input_ov(), infer_completion_callback(), infer_completion_callback(), infer_completion_callback(), and infer_completion_callback().

dnn_get_channel_idx_by_layout()

static int dnn_get_channel_idx_by_layout ( DNNLayout layout )

inlinestatic

Definition at line 219 of file dnn_interface.h.

Referenced by allocate_input_tensor(), check_modelinput_inlink(), fill_model_input_onnx(), fill_model_input_th(), and infer_completion_callback().