summaryrefslogtreecommitdiff
path: root/libavfilter/dnn/dnn_backend_common.h
diff options
context:
space:
mode:
Diffstat (limited to 'libavfilter/dnn/dnn_backend_common.h')
-rw-r--r--libavfilter/dnn/dnn_backend_common.h19
1 files changed, 19 insertions, 0 deletions
diff --git a/libavfilter/dnn/dnn_backend_common.h b/libavfilter/dnn/dnn_backend_common.h
index cd9c0f5339..0c043e51f0 100644
--- a/libavfilter/dnn/dnn_backend_common.h
+++ b/libavfilter/dnn/dnn_backend_common.h
@@ -26,6 +26,25 @@
#include "../dnn_interface.h"
+// one task for one function call from dnn interface
+typedef struct TaskItem {
+ void *model; // model for the backend
+ AVFrame *in_frame;
+ AVFrame *out_frame;
+ const char *input_name;
+ const char *output_name;
+ int async;
+ int do_ioproc;
+ uint32_t inference_todo;
+ uint32_t inference_done;
+} TaskItem;
+
+// one task might have multiple inferences
+typedef struct InferenceItem {
+ TaskItem *task;
+ uint32_t bbox_index;
+} InferenceItem;
+
int ff_check_exec_params(void *ctx, DNNBackendType backend, DNNFunctionType func_type, DNNExecBaseParams *exec_params);
#endif