diff options
| author | Sergey Lavrushkin <dualfal@gmail.com> | 2018-06-03 20:22:50 +0300 |
|---|---|---|
| committer | Pedro Arthur <bygrandao@gmail.com> | 2018-06-05 10:16:50 -0300 |
| commit | d8c0bbb0aa45eed61b159c4842473fe27e77ac12 (patch) | |
| tree | cc3f2b09aacf93ea7d5ce42adfd62a4fcc561fb1 /libavfilter/dnn_interface.c | |
| parent | ddf6ff9dc6e6441b68b83b3bf047dbc573cc5b7c (diff) | |
| download | ffmpeg-d8c0bbb0aa45eed61b159c4842473fe27e77ac12.tar.gz | |
Adds TensorFlow backend for dnn inference module.
Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
Diffstat (limited to 'libavfilter/dnn_interface.c')
| -rw-r--r-- | libavfilter/dnn_interface.c | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/libavfilter/dnn_interface.c b/libavfilter/dnn_interface.c index e981829b95..ce64fe58f3 100644 --- a/libavfilter/dnn_interface.c +++ b/libavfilter/dnn_interface.c @@ -25,6 +25,7 @@ #include "dnn_interface.h" #include "dnn_backend_native.h" +#include "dnn_backend_tf.h" #include "libavutil/mem.h" DNNModule* ff_get_dnn_module(DNNBackendType backend_type) @@ -42,6 +43,17 @@ DNNModule* ff_get_dnn_module(DNNBackendType backend_type) dnn_module->load_default_model = &ff_dnn_load_default_model_native; dnn_module->execute_model = &ff_dnn_execute_model_native; dnn_module->free_model = &ff_dnn_free_model_native; + break; + case DNN_TF: + #if (CONFIG_LIBTENSORFLOW == 1) + dnn_module->load_model = &ff_dnn_load_model_tf; + dnn_module->load_default_model = &ff_dnn_load_default_model_tf; + dnn_module->execute_model = &ff_dnn_execute_model_tf; + dnn_module->free_model = &ff_dnn_free_model_tf; + #else + av_freep(dnn_module); + return NULL; + #endif } return dnn_module; |