videolan
/
ffmpeg
mirror of https://git.videolan.org/git/ffmpeg.git
1
Fork 0
Code Releases Activity

avfilter/dnn: add log context to ff_get_dnn_module 

Print backend type when failed.

Signed-off-by: Zhao Zhili <zhilizhao@tencent.com>
This commit is contained in:
Zhao Zhili 2023-04-30 23:38:55 +08:00
parent 505c43bb65
commit 016f2f61c3
3 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libavfilter/dnn/dnn_interface.c
View File

@ -29,7 +29,7 @@
extern const DNNModule ff_dnn_backend_openvino;
extern const DNNModule ff_dnn_backend_tf;
const DNNModule *ff_get_dnn_module(DNNBackendType backend_type)
const DNNModule *ff_get_dnn_module(DNNBackendType backend_type, void *log_ctx)
{
switch(backend_type){
#if (CONFIG_LIBTENSORFLOW == 1)
@ -41,7 +41,9 @@ const DNNModule *ff_get_dnn_module(DNNBackendType backend_type)
return &ff_dnn_backend_openvino;
#endif
default:
av_log(NULL, AV_LOG_ERROR, "Module backend_type is not supported or enabled.\n");
av_log(log_ctx, AV_LOG_ERROR,
"Module backend_type %d is not supported or enabled.\n",
backend_type);
return NULL;
}
}

2
libavfilter/dnn_filter_common.c
View File

@ -68,7 +68,7 @@ int ff_dnn_init(DnnContext *ctx, DNNFunctionType func_type, AVFilterContext *fil
return AVERROR(EINVAL);
}
ctx->dnn_module = ff_get_dnn_module(ctx->backend_type);
ctx->dnn_module = ff_get_dnn_module(ctx->backend_type, filter_ctx);
if (!ctx->dnn_module) {
av_log(filter_ctx, AV_LOG_ERROR, "could not create DNN module for requested backend\n");
return AVERROR(ENOMEM);

2
libavfilter/dnn_interface.h
View File

@ -123,6 +123,6 @@ typedef struct DNNModule{
} DNNModule;
// Initializes DNNModule depending on chosen backend.
const DNNModule *ff_get_dnn_module(DNNBackendType backend_type);
const DNNModule *ff_get_dnn_module(DNNBackendType backend_type, void *log_ctx);
#endif