[FFmpeg-devel] [PATCH v2 FFmpeg 7/20] libavfilter/dnn_interface.h: define new DNNExecParams DNNExecZeroShotClassificationParams

[FFmpeg-devel] [PATCH v2 FFmpeg 7/20] libavfilter/dnn_interface.h: define new DNNExecParams DNNExecZeroShotClassificationParams

[m.kaindl0208]

Signed-off-by: MaximilianKaindl <m.kaindl0208@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-request@ffmpeg.org with subject "unsubscribe".