summaryrefslogtreecommitdiff
path: root/libavfilter/dnn_filter_common.h
diff options
context:
space:
mode:
authorGuo, Yejun <yejun.guo@intel.com>2021-03-01 19:23:20 +0800
committerGuo, Yejun <yejun.guo@intel.com>2021-04-08 09:23:02 +0800
commit59021d79a24e28434f57376276625bc44eff340c (patch)
treeb67d8f2e1b0b7697bd6e3e39a931652917084c2f /libavfilter/dnn_filter_common.h
parentd2ccbc966b151f2500aa43e789288af188258ba3 (diff)
downloadffmpeg-59021d79a24e28434f57376276625bc44eff340c.tar.gz
lavfi/dnn: refine code for frame pre/proc processing
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 79c4d3efe3..c611d594dc 100644
--- a/libavfilter/dnn_filter_common.h
+++ b/libavfilter/dnn_filter_common.h
@@ -48,6 +48,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);
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);