More support for pose estimation in yolo-v8. (#599)

* More support for pose estimation in yolo-v8.

* Support both object detection and pose-estimation in the yolo-v8 example.
This commit is contained in:
Laurent Mazare
2023-08-25 11:21:11 +01:00
committed by GitHub
parent afc10a3232
commit 8bc5fffa45
3 changed files with 164 additions and 16 deletions

View File

@ -65,6 +65,7 @@ pub fn report(
xmax: pred[0] + pred[2] / 2.,
ymax: pred[1] + pred[3] / 2.,
confidence,
keypoints: vec![],
};
bboxes[class_index].push(bbox)
}