summaryrefslogtreecommitdiff
path: root/libavfilter/dnn_interface.h
diff options
context:
space:
mode:
authorSergey Lavrushkin <dualfal@gmail.com>2018-06-14 00:37:12 +0300
committerPedro Arthur <bygrandao@gmail.com>2018-07-02 10:47:14 -0300
commit575b7189908e1cfa55104b0d2c7c9f6ea30ca2dc (patch)
tree49ba8795536d104ec1d1c8cb08772fcceb8da431 /libavfilter/dnn_interface.h
parentd24c9e55f64eebf67a9e488daa17332533481c20 (diff)
Adds ESPCN super resolution filter merged with SRCNN filter.
Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
Diffstat (limited to 'libavfilter/dnn_interface.h')
-rw-r--r--libavfilter/dnn_interface.h4
1 files changed, 2 insertions, 2 deletions
diff --git a/libavfilter/dnn_interface.h b/libavfilter/dnn_interface.h
index 909ba95edf..6b820d1d5b 100644
--- a/libavfilter/dnn_interface.h
+++ b/libavfilter/dnn_interface.h
@@ -30,7 +30,7 @@ typedef enum {DNN_SUCCESS, DNN_ERROR} DNNReturnType;
typedef enum {DNN_NATIVE, DNN_TF} DNNBackendType;
-typedef enum {DNN_SRCNN} DNNDefaultModel;
+typedef enum {DNN_SRCNN, DNN_ESPCN} DNNDefaultModel;
typedef struct DNNData{
float* data;
@@ -42,7 +42,7 @@ typedef struct DNNModel{
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);
+ DNNReturnType (*set_input_output)(void* model, DNNData* input, DNNData* output);
} DNNModel;
// Stores pointers to functions for loading, executing, freeing DNN models for one of the backends.