FFmpeg
|
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 | |
DNNModel * | ff_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, const char *input_name, AVFrame *in_frame, const char **output_names, uint32_t nb_output, AVFrame *out_frame) |
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) |
DNN inference functions interface for native backend.
Definition in file dnn_backend_native.h.
enum 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 39 of file dnn_backend_native.h.
enum DNNOperandType |
Enumerator | |
---|---|
DOT_INPUT | |
DOT_OUTPUT | |
DOT_INTERMEDIATE |
Definition at line 52 of file dnn_backend_native.h.
enum DNNPaddingParam |
Enumerator | |
---|---|
VALID | |
SAME | |
SAME_CLAMP_TO_EDGE |
Definition at line 53 of file dnn_backend_native.h.
enum DNNActivationFunc |
Enumerator | |
---|---|
RELU | |
TANH | |
SIGMOID | |
NONE | |
LEAKY_RELU |
Definition at line 54 of file dnn_backend_native.h.
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 115 of file dnn_backend_native.c.
Referenced by ff_get_dnn_module(), and load_native_model().
DNNReturnType ff_dnn_execute_model_native | ( | const DNNModel * | model, |
const char * | input_name, | ||
AVFrame * | in_frame, | ||
const char ** | output_names, | ||
uint32_t | nb_output, | ||
AVFrame * | out_frame | ||
) |
Definition at line 375 of file dnn_backend_native.c.
Referenced by ff_get_dnn_module().
void ff_dnn_free_model_native | ( | DNNModel ** | model | ) |
Definition at line 415 of file dnn_backend_native.c.
Referenced by ff_dnn_load_model_native(), ff_get_dnn_module(), and load_native_model().
int32_t ff_calculate_operand_data_length | ( | const DnnOperand * | oprd | ) |
Definition at line 403 of file dnn_backend_native.c.
Referenced by execute_model_native(), ff_dnn_execute_layer_avg_pool(), ff_dnn_execute_layer_conv2d(), ff_dnn_execute_layer_dense(), ff_dnn_execute_layer_depth2space(), ff_dnn_execute_layer_math_binary(), ff_dnn_execute_layer_math_unary(), ff_dnn_execute_layer_maximum(), and ff_dnn_execute_layer_pad().
int32_t ff_calculate_operand_dims_count | ( | const DnnOperand * | oprd | ) |
Definition at line 394 of file dnn_backend_native.c.
Referenced by ff_dnn_execute_layer_math_unary(), ff_dnn_execute_layer_maximum(), math_binary_commutative(), and math_binary_not_commutative().