From e2545131df80604fcd6f981cf44167b7b5da4994 Mon Sep 17 00:00:00 2001
From: lindsayshuo <932695342@qq.com>
Date: Wed, 28 Aug 2024 17:28:13 +0800
Subject: [PATCH] keypoints array with dynamic size based on kNumberOfPoints
 (#1570)

* Add the generation of multi-class pose engines

* Change grids in forwardGpu to one-dimensional arrays

* Update README.md

* Update types.h

keypoints array with dynamic size based on kNumberOfPoints

---------

Co-authored-by: lindsayshuo <lindsayshuo@foxmail.com.com>
---
 yolov8/include/types.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/yolov8/include/types.h b/yolov8/include/types.h
index 472c735..c43f589 100644
--- a/yolov8/include/types.h
+++ b/yolov8/include/types.h
@@ -7,7 +7,7 @@ struct alignas(float) Detection {
     float conf;  // bbox_conf * cls_conf
     float class_id;
     float mask[32];
-    float keypoints[51];  // 17*3 keypoints
+    float keypoints[kNumberOfPoints * 3];  // keypoints array with dynamic size based on kNumberOfPoints
 };
 
 struct AffineMatrix {