summaryrefslogtreecommitdiff
path: root/libavfilter/dnn_interface.h
diff options
context:
space:
mode:
authorSergey Lavrushkin <dualfal@gmail.com>2018-06-03 20:22:50 +0300
committerPedro Arthur <bygrandao@gmail.com>2018-06-05 10:16:50 -0300
commitd8c0bbb0aa45eed61b159c4842473fe27e77ac12 (patch)
treecc3f2b09aacf93ea7d5ce42adfd62a4fcc561fb1 /libavfilter/dnn_interface.h
parentddf6ff9dc6e6441b68b83b3bf047dbc573cc5b7c (diff)
Adds TensorFlow backend for dnn inference module.
Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
Diffstat (limited to 'libavfilter/dnn_interface.h')
-rw-r--r--libavfilter/dnn_interface.h3
1 files changed, 1 insertions, 2 deletions
diff --git a/libavfilter/dnn_interface.h b/libavfilter/dnn_interface.h
index 52e344cee4..909ba95edf 100644
--- a/libavfilter/dnn_interface.h
+++ b/libavfilter/dnn_interface.h
@@ -28,7 +28,7 @@
typedef enum {DNN_SUCCESS, DNN_ERROR} DNNReturnType;
-typedef enum {DNN_NATIVE} DNNBackendType;
+typedef enum {DNN_NATIVE, DNN_TF} DNNBackendType;
typedef enum {DNN_SRCNN} DNNDefaultModel;
@@ -37,7 +37,6 @@ typedef struct DNNData{
int width, height, channels;
} DNNData;
-
typedef struct DNNModel{
// Stores model that can be different for different backends.
void* model;