diff options
author | Guo, Yejun <yejun.guo@intel.com> | 2021-03-09 09:51:42 +0300 |
---|---|---|
committer | Guo, Yejun <yejun.guo@intel.com> | 2021-04-08 04:23:02 +0300 |
commit | 13bf797ced0b527fa770d4b29884a5b0b8f19898 (patch) | |
tree | e7ea6390670e4d124c7beaa344074184d5d009d0 /libavfilter/dnn_filter_common.c | |
parent | 59021d79a24e28434f57376276625bc44eff340c (diff) |
lavfi/dnn: add post process for detection
Diffstat (limited to 'libavfilter/dnn_filter_common.c')
-rw-r--r-- | libavfilter/dnn_filter_common.c | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/libavfilter/dnn_filter_common.c b/libavfilter/dnn_filter_common.c index dc5966332a..1b922455a3 100644 --- a/libavfilter/dnn_filter_common.c +++ b/libavfilter/dnn_filter_common.c @@ -71,6 +71,12 @@ int ff_dnn_set_frame_proc(DnnContext *ctx, FramePrePostProc pre_proc, FramePrePo return 0; } +int ff_dnn_set_detect_post_proc(DnnContext *ctx, DetectPostProc post_proc) +{ + ctx->model->detect_post_proc = post_proc; + return 0; +} + DNNReturnType ff_dnn_get_input(DnnContext *ctx, DNNData *input) { return ctx->model->get_input(ctx->model->model, input, ctx->model_inputname); |