virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-08-10 06:10:52 +02:00
Code Issues Releases Activity

avfilter/dnn: refactor ff_get_dnn_module to remove allocation

Browse Source 
Signed-off-by: Zhao Zhili <zhilizhao@tencent.com>
This commit is contained in:
Zhao Zhili
2023-04-30 23:38:54 +08:00
parent 3f52b7eedc
commit 505c43bb65
4 changed files with 7 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
libavfilter/dnn/dnn_interface.c
View File

@@ -29,37 +29,19 @@
extern const DNNModule ff_dnn_backend_openvino;
extern const DNNModule ff_dnn_backend_tf;
DNNModule *ff_get_dnn_module(DNNBackendType backend_type)
const DNNModule *ff_get_dnn_module(DNNBackendType backend_type)
{
DNNModule *dnn_module;
dnn_module = av_mallocz(sizeof(DNNModule));
if(!dnn_module){
return NULL;
}
switch(backend_type){
case DNN_TF:
#if (CONFIG_LIBTENSORFLOW == 1)
*dnn_module = ff_dnn_backend_tf;
#else
av_freep(&dnn_module);
return NULL;
case DNN_TF:
return &ff_dnn_backend_tf;
#endif
break;
case DNN_OV:
#if (CONFIG_LIBOPENVINO == 1)
*dnn_module = ff_dnn_backend_openvino;
#else
av_freep(&dnn_module);
return NULL;
case DNN_OV:
return &ff_dnn_backend_openvino;
#endif
break;
default:
av_log(NULL, AV_LOG_ERROR, "Module backend_type is not supported or enabled.\n");
av_freep(&dnn_module);
return NULL;
}
return dnn_module;
}