FFmpeg
Data Structures | Enumerations | Functions
dnn_backend_native.h File Reference
#include "../dnn_interface.h"
#include "libavformat/avio.h"
#include "libavutil/opt.h"
#include "queue.h"

Go to the source code of this file.

Data Structures

struct  Layer
 
struct  DnnOperand
 
struct  InputParams
 
struct  NativeOptions
 
struct  NativeContext
 
struct  NativeModel
 

Enumerations

enum  DNNLayerType {
  DLT_INPUT = 0, DLT_CONV2D = 1, DLT_DEPTH_TO_SPACE = 2, DLT_MIRROR_PAD = 3,
  DLT_MAXIMUM = 4, DLT_MATH_BINARY = 5, DLT_MATH_UNARY = 6, DLT_AVG_POOL = 7,
  DLT_DENSE = 8, DLT_COUNT
}
 the enum value of DNNLayerType should not be changed, the same values are used in convert_from_tensorflow.py and, it is used to index the layer execution/load function pointer. More...
 
enum  DNNOperandType { DOT_INPUT = 1, DOT_OUTPUT = 2, DOT_INTERMEDIATE = DOT_INPUT | DOT_OUTPUT }
 
enum  DNNPaddingParam { VALID, SAME, SAME_CLAMP_TO_EDGE }
 
enum  DNNActivationFunc {
  RELU, TANH, SIGMOID, NONE,
  LEAKY_RELU
}
 

Functions

DNNModelff_dnn_load_model_native (const char *model_filename, DNNFunctionType func_type, const char *options, AVFilterContext *filter_ctx)
 
DNNReturnType ff_dnn_execute_model_native (const DNNModel *model, DNNExecBaseParams *exec_params)
 
DNNAsyncStatusType ff_dnn_get_result_native (const DNNModel *model, AVFrame **in, AVFrame **out)
 
DNNReturnType ff_dnn_flush_native (const DNNModel *model)
 
void ff_dnn_free_model_native (DNNModel **model)
 
int32_t ff_calculate_operand_data_length (const DnnOperand *oprd)
 
int32_t ff_calculate_operand_dims_count (const DnnOperand *oprd)
 

Detailed Description

DNN inference functions interface for native backend.

Definition in file dnn_backend_native.h.

Enumeration Type Documentation

◆ DNNLayerType

the enum value of DNNLayerType should not be changed, the same values are used in convert_from_tensorflow.py and, it is used to index the layer execution/load function pointer.

Enumerator
DLT_INPUT 
DLT_CONV2D 
DLT_DEPTH_TO_SPACE 
DLT_MIRROR_PAD 
DLT_MAXIMUM 
DLT_MATH_BINARY 
DLT_MATH_UNARY 
DLT_AVG_POOL 
DLT_DENSE 
DLT_COUNT 

Definition at line 40 of file dnn_backend_native.h.

◆ DNNOperandType

Enumerator
DOT_INPUT 
DOT_OUTPUT 
DOT_INTERMEDIATE 

Definition at line 53 of file dnn_backend_native.h.

◆ DNNPaddingParam

Enumerator
VALID 
SAME 
SAME_CLAMP_TO_EDGE 

Definition at line 54 of file dnn_backend_native.h.

◆ DNNActivationFunc

Enumerator
RELU 
TANH 
SIGMOID 
NONE 
LEAKY_RELU 

Definition at line 55 of file dnn_backend_native.h.

Function Documentation

◆ ff_dnn_load_model_native()

DNNModel* ff_dnn_load_model_native ( const char *  model_filename,
DNNFunctionType  func_type,
const char *  options,
AVFilterContext filter_ctx 
)

check file header with string and version

Definition at line 139 of file dnn_backend_native.c.

Referenced by ff_get_dnn_module(), and load_native_model().

◆ ff_dnn_execute_model_native()

DNNReturnType ff_dnn_execute_model_native ( const DNNModel model,
DNNExecBaseParams exec_params 
)

Definition at line 435 of file dnn_backend_native.c.

Referenced by ff_get_dnn_module().

◆ ff_dnn_get_result_native()

DNNAsyncStatusType ff_dnn_get_result_native ( const DNNModel model,
AVFrame **  in,
AVFrame **  out 
)

Definition at line 484 of file dnn_backend_native.c.

Referenced by ff_get_dnn_module().

◆ ff_dnn_flush_native()

DNNReturnType ff_dnn_flush_native ( const DNNModel model)

Definition at line 470 of file dnn_backend_native.c.

Referenced by ff_get_dnn_module().

◆ ff_dnn_free_model_native()

void ff_dnn_free_model_native ( DNNModel **  model)

◆ ff_calculate_operand_data_length()

int32_t ff_calculate_operand_data_length ( const DnnOperand oprd)

◆ ff_calculate_operand_dims_count()

int32_t ff_calculate_operand_dims_count ( const DnnOperand oprd)