diff options
| author | Guo, Yejun <yejun.guo@intel.com> | 2021-03-01 19:23:20 +0800 |
|---|---|---|
| committer | Guo, Yejun <yejun.guo@intel.com> | 2021-04-08 09:23:02 +0800 |
| commit | 59021d79a24e28434f57376276625bc44eff340c (patch) | |
| tree | b67d8f2e1b0b7697bd6e3e39a931652917084c2f /libavfilter/dnn/dnn_backend_native.c | |
| parent | d2ccbc966b151f2500aa43e789288af188258ba3 (diff) | |
lavfi/dnn: refine code for frame pre/proc processing
Diffstat (limited to 'libavfilter/dnn/dnn_backend_native.c')
| -rw-r--r-- | libavfilter/dnn/dnn_backend_native.c | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/libavfilter/dnn/dnn_backend_native.c b/libavfilter/dnn/dnn_backend_native.c index d8ae36c52d..d9762eeaf6 100644 --- a/libavfilter/dnn/dnn_backend_native.c +++ b/libavfilter/dnn/dnn_backend_native.c @@ -310,8 +310,8 @@ static DNNReturnType execute_model_native(const DNNModel *model, const char *inp input.data = oprd->data; input.dt = oprd->data_type; if (do_ioproc) { - if (native_model->model->pre_proc != NULL) { - native_model->model->pre_proc(in_frame, &input, native_model->model->filter_ctx); + if (native_model->model->frame_pre_proc != NULL) { + native_model->model->frame_pre_proc(in_frame, &input, native_model->model->filter_ctx); } else { ff_proc_from_frame_to_dnn(in_frame, &input, native_model->model->func_type, ctx); } @@ -358,8 +358,8 @@ static DNNReturnType execute_model_native(const DNNModel *model, const char *inp output.dt = oprd->data_type; if (do_ioproc) { - if (native_model->model->post_proc != NULL) { - native_model->model->post_proc(out_frame, &output, native_model->model->filter_ctx); + if (native_model->model->frame_post_proc != NULL) { + native_model->model->frame_post_proc(out_frame, &output, native_model->model->filter_ctx); } else { ff_proc_from_dnn_to_frame(out_frame, &output, ctx); } |