Commit 4b003322 authored by Steven Liu's avatar Steven Liu

avfilter/sr: process and output message when load_model is NULL

fix ticket: 7455
Reviewed-by: 's avatarPedro Arthur <bygrandao@gmail.com>
Reviewed-by: 's avatarPaul B Mahol <onemda@gmail.com>
Signed-off-by: 's avatarSteven Liu <lq@chinaffmpeg.org>
parent 8d299307
...@@ -52,6 +52,11 @@ DNNModule *ff_get_dnn_module(DNNBackendType backend_type) ...@@ -52,6 +52,11 @@ DNNModule *ff_get_dnn_module(DNNBackendType backend_type)
av_freep(&dnn_module); av_freep(&dnn_module);
return NULL; return NULL;
#endif #endif
break;
default:
av_log(NULL, AV_LOG_ERROR, "Module backend_type is not native or tensorflow\n");
av_freep(&dnn_module);
return NULL;
} }
return dnn_module; return dnn_module;
......
...@@ -75,8 +75,13 @@ static av_cold int init(AVFilterContext *context) ...@@ -75,8 +75,13 @@ static av_cold int init(AVFilterContext *context)
return AVERROR(EIO); return AVERROR(EIO);
} }
else{ else{
if (!sr_context->dnn_module->load_model) {
av_log(context, AV_LOG_ERROR, "load_model for network was not specified\n");
return AVERROR(EIO);
} else {
sr_context->model = (sr_context->dnn_module->load_model)(sr_context->model_filename); sr_context->model = (sr_context->dnn_module->load_model)(sr_context->model_filename);
} }
}
if (!sr_context->model){ if (!sr_context->model){
av_log(context, AV_LOG_ERROR, "could not load DNN model\n"); av_log(context, AV_LOG_ERROR, "could not load DNN model\n");
return AVERROR(EIO); return AVERROR(EIO);
......
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