haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

keypoints array with dynamic size based on kNumberOfPoints (#1570)

Browse Source 
* 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>
This commit is contained in:
lindsayshuo 2024-08-28 17:28:13 +08:00 committed by GitHub
parent d5230129c8
commit e2545131df
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yolov8/include/types.h
View File

@ -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 {