mirror of
https://github.com/huggingface/candle.git
synced 2025-06-15 02:16:37 +00:00
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:
@ -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)
|
||||
}
|
||||
|
Reference in New Issue
Block a user