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

Move the yolo model bits in a separate file. (#602)

Browse Source 
* Move the yolo model bits in a separate file.

* Improve the drawing.

* Bugfix.
This commit is contained in:
Laurent Mazare
2023-08-25 12:47:55 +01:00
committed by GitHub
parent 8bc5fffa45
commit 97909e5068
5 changed files with 806 additions and 749 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,7 +6,14 @@ pub struct Bbox {
pub xmax: f32,
pub ymax: f32,
pub confidence: f32,
pub keypoints: Vec<(f32, f32, f32)>,
pub keypoints: Vec<KeyPoint>,
}
#[derive(Debug, Clone, Copy, PartialEq)]
pub struct KeyPoint {
pub x: f32,
pub y: f32,
pub mask: f32,
}
/// Intersection over union of two bounding boxes.