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)
av_freep(&dnn_module);
return NULL;
#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;
......
......@@ -75,8 +75,13 @@ static av_cold int init(AVFilterContext *context)
return AVERROR(EIO);
}
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);
}
}
if (!sr_context->model){
av_log(context, AV_LOG_ERROR, "could not load DNN model\n");
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