diff options
author | Sergey Lavrushkin <dualfal@gmail.com> | 2018-05-25 20:31:04 +0300 |
---|---|---|
committer | Pedro Arthur <bygrandao@gmail.com> | 2018-05-29 10:02:30 -0300 |
commit | bdf1bbdbb4ebb342c0267d0f77cd06e717197e65 (patch) | |
tree | 9dcfad8953ce29cc74d8f152ee8c462cf4b72fe7 /libavfilter/dnn_interface.h | |
parent | cba167934bb2d634f400d3b06cec2b0f99d5de9d (diff) |
Adds dnn inference module for simple convolutional networks. Reimplements srcnn filter based on it.
Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
Diffstat (limited to 'libavfilter/dnn_interface.h')
-rw-r--r-- | libavfilter/dnn_interface.h | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/libavfilter/dnn_interface.h b/libavfilter/dnn_interface.h new file mode 100644 index 0000000000..52e344cee4 --- /dev/null +++ b/libavfilter/dnn_interface.h @@ -0,0 +1,64 @@ +/* + * Copyright (c) 2018 Sergey Lavrushkin + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +/** + * @file + * DNN inference engine interface. + */ + +#ifndef AVFILTER_DNN_INTERFACE_H +#define AVFILTER_DNN_INTERFACE_H + +typedef enum {DNN_SUCCESS, DNN_ERROR} DNNReturnType; + +typedef enum {DNN_NATIVE} DNNBackendType; + +typedef enum {DNN_SRCNN} DNNDefaultModel; + +typedef struct DNNData{ + float* data; + int width, height, channels; +} DNNData; + + +typedef struct DNNModel{ + // Stores model that can be different for different backends. + void* model; + // Sets model input and output, while allocating additional memory for intermediate calculations. + // Should be called at least once before model execution. + DNNReturnType (*set_input_output)(void* model, const DNNData* input, const DNNData* output); +} DNNModel; + +// Stores pointers to functions for loading, executing, freeing DNN models for one of the backends. +typedef struct DNNModule{ + // Loads model and parameters from given file. Returns NULL if it is not possible. + DNNModel* (*load_model)(const char* model_filename); + // Loads one of the default models + DNNModel* (*load_default_model)(DNNDefaultModel model_type); + // Executes model with specified input and output. Returns DNN_ERROR otherwise. + DNNReturnType (*execute_model)(const DNNModel* model); + // Frees memory allocated for model. + void (*free_model)(DNNModel** model); +} DNNModule; + +// Initializes DNNModule depending on chosen backend. +DNNModule* ff_get_dnn_module(DNNBackendType backend_type); + +#endif |