servius/face-detector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: output aabbs from retinaface

Browse Source
This commit is contained in:
uttarayan21
2025-08-04 11:43:16 +05:30
parent c751654799
commit f55f0ab089
4 changed files with 220 additions and 182 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/main.rs
View File

@@ -14,7 +14,7 @@ pub fn main() -> Result<()> {
match args.cmd {
cli::SubCommand::Detect(detect) => {
use detector::facedet;
let model = facedet::FaceDetection::new_from_bytes(RETINAFACE_MODEL)
let model = facedet::retinaface::FaceDetection::new_from_bytes(RETINAFACE_MODEL)
.change_context(errors::Error)
.attach_printable("Failed to create face detection model")?;
let image = image::open(detect.image).change_context(Error)?;
@@ -27,7 +27,9 @@ pub fn main() -> Result<()> {
.detect_faces(array)
.change_context(errors::Error)
.attach_printable("Failed to detect faces")?;
output.print(20);
// output.print(20);
let aabbs = output.postprocess(Default::default());
dbg!(aabbs);
}
cli::SubCommand::List(list) => {
println!("List: {:?}", list);