Signed-off-by: MaximilianKaindl <m.kaindl0...@gmail.com>
---
libavfilter/dnn_interface.h | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/libavfilter/dnn_interface.h b/libavfilter/dnn_interface.h
index f4552d4287..4252cd2231 100644
--- a/libavfilter/dnn_interface.h
+++ b/libavfilter/dnn_interface.h
@@ -92,6 +92,16 @@ typedef struct DNNExecClassificationParams {
const char *target;
} DNNExecClassificationParams;
+typedef struct DNNExecZeroShotClassificationParams {
+ DNNExecBaseParams base;
+ const char **labels;
+ const int label_count;
+ const char *target;
+ const char *tokenizer_path;
+ const int *softmax_units;
+ const int softmax_units_count;
+} DNNExecZeroShotClassificationParams;
+
typedef int (*FramePrePostProc)(AVFrame *frame, DNNData *model,
AVFilterContext *filter_ctx);
typedef int (*DetectPostProc)(AVFrame *frame, DNNData *output, uint32_t nb,
AVFilterContext *filter_ctx);
typedef int (*ClassifyPostProc)(AVFrame *frame, DNNData *output, uint32_t
bbox_index, AVFilterContext *filter_ctx);
--
2.34.1
_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".
