Commit 016f2f61 authored by Zhao Zhili's avatar Zhao Zhili

avfilter/dnn: add log context to ff_get_dnn_module

Print backend type when failed.
Signed-off-by: 's avatarZhao Zhili <zhilizhao@tencent.com>
parent 505c43bb
......@@ -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;
}
}
......@@ -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);
......
......@@ -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
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment