apps/ffmpeg-rockchip
1
0
Fork 0
mirror of https://github.com/nyanmisaka/ffmpeg-rockchip.git synced 2025-11-02 13:24:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adds TensorFlow backend for dnn inference module.

Browse Source 
Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
This commit is contained in:
Sergey Lavrushkin
2018-06-03 20:22:50 +03:00
committed by Pedro Arthur
parent ddf6ff9dc6
commit d8c0bbb0aa
9 changed files with 3262 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
libavfilter/dnn_interface.c
View File

@@ -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;