huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 19:18:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-examples/src/object_detection.rs
View File

@ -1,11 +1,12 @@
/// A bounding box around an object.
#[derive(Debug, Clone, Copy)]
#[derive(Debug, Clone)]
pub struct Bbox {
pub xmin: f32,
pub ymin: f32,
pub xmax: f32,
pub ymax: f32,
pub confidence: f32,
pub keypoints: Vec<(f32, f32, f32)>,
}
/// Intersection over union of two bounding boxes.