diff options
author | Guo, Yejun <yejun.guo@intel.com> | 2021-02-07 14:35:22 +0800 |
---|---|---|
committer | Guo, Yejun <yejun.guo@intel.com> | 2021-02-18 09:59:37 +0800 |
commit | 76fc6879e2c59929e7d81a72986bb4950053177e (patch) | |
tree | 62ac1a7bd345bfe6fd06f15113b1d46e466413ee /libavfilter/dnn_filter_common.h | |
parent | bdce6361004b3a24082e00b6c0126e2694a65b50 (diff) | |
download | ffmpeg-76fc6879e2c59929e7d81a72986bb4950053177e.tar.gz |
dnn: add function type for model
So the backend knows the usage of model is for frame processing,
detect, classify, etc. Each function type has different behavior
in backend when handling the input/output data of the model.
Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
Diffstat (limited to 'libavfilter/dnn_filter_common.h')
-rw-r--r-- | libavfilter/dnn_filter_common.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/libavfilter/dnn_filter_common.h b/libavfilter/dnn_filter_common.h index ab49a992ed..79c4d3efe3 100644 --- a/libavfilter/dnn_filter_common.h +++ b/libavfilter/dnn_filter_common.h @@ -47,7 +47,7 @@ typedef struct DnnContext { { "async", "use DNN async inference", OFFSET(async), AV_OPT_TYPE_BOOL, { .i64 = 1}, 0, 1, FLAGS}, -int ff_dnn_init(DnnContext *ctx, AVFilterContext *filter_ctx); +int ff_dnn_init(DnnContext *ctx, DNNFunctionType func_type, AVFilterContext *filter_ctx); 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); |