summaryrefslogtreecommitdiff
path: root/libavfilter/dnn_filter_common.h
diff options
context:
space:
mode:
authorGuo, Yejun <yejun.guo@intel.com>2021-03-09 14:51:42 +0800
committerGuo, Yejun <yejun.guo@intel.com>2021-04-08 09:23:02 +0800
commit13bf797ced0b527fa770d4b29884a5b0b8f19898 (patch)
treee7ea6390670e4d124c7beaa344074184d5d009d0 /libavfilter/dnn_filter_common.h
parent59021d79a24e28434f57376276625bc44eff340c (diff)
downloadffmpeg-13bf797ced0b527fa770d4b29884a5b0b8f19898.tar.gz
lavfi/dnn: add post process for detection
Diffstat (limited to 'libavfilter/dnn_filter_common.h')
-rw-r--r--libavfilter/dnn_filter_common.h1
1 files changed, 1 insertions, 0 deletions
diff --git a/libavfilter/dnn_filter_common.h b/libavfilter/dnn_filter_common.h
index c611d594dc..8deb18b39a 100644
--- a/libavfilter/dnn_filter_common.h
+++ b/libavfilter/dnn_filter_common.h
@@ -49,6 +49,7 @@ typedef struct DnnContext {
int ff_dnn_init(DnnContext *ctx, DNNFunctionType func_type, AVFilterContext *filter_ctx);
int ff_dnn_set_frame_proc(DnnContext *ctx, FramePrePostProc pre_proc, FramePrePostProc post_proc);
+int ff_dnn_set_detect_post_proc(DnnContext *ctx, DetectPostProc post_proc);
DNNReturnType ff_dnn_get_input(DnnContext *ctx, DNNData *input);
DNNReturnType ff_dnn_get_output(DnnContext *ctx, int input_width, int input_height, int *output_width, int *output_height);
DNNReturnType ff_dnn_execute_model(DnnContext *ctx, AVFrame *in_frame, AVFrame *out_frame);