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

feat: Added postprocessing for retinaface

Browse Source
This commit is contained in:
uttarayan21
2025-08-04 19:27:45 +05:30
parent f55f0ab089
commit df5584d797
8 changed files with 235 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
src/main.rs
View File

@@ -19,17 +19,45 @@ pub fn main() -> Result<()> {
.attach_printable("Failed to create face detection model")?;
let image = image::open(detect.image).change_context(Error)?;
let image = image.into_rgb8();
let array = image
let mut array = image
.into_ndarray()
.change_context(errors::Error)
.attach_printable("Failed to convert image to ndarray")?;
let output = model
.detect_faces(array)
.detect_faces(array.clone())
.change_context(errors::Error)
.attach_printable("Failed to detect faces")?;
// output.print(20);
let aabbs = output.postprocess(Default::default());
dbg!(aabbs);
let aabbs = output
.postprocess(Default::default())
.change_context(errors::Error)
.attach_printable("Failed to attach context")?;
for bbox in aabbs {
println!("Detected face: {:?}", bbox);
use bounding_box::draw::*;
let bbox = bbox
.denormalize(nalgebra::SVector::<f32, 2>::new(
array.shape()[1] as f32,
array.shape()[0] as f32,
))
.cast()
.ok_or(errors::Error)
.attach_printable("Failed to cast f32 to usize")?;
dbg!(bbox);
array.draw(bbox, color::palette::css::GREEN_YELLOW.to_rgba8(), 20);
break;
}
let v = array.view();
if let Some(output) = detect.output {
let image: image::RgbImage = v
.to_image()
.change_context(errors::Error)
.attach_printable("Failed to convert ndarray to image")?;
image
.save(output)
.change_context(errors::Error)
.attach_printable("Failed to save output image")?;
}
}
cli::SubCommand::List(list) => {
println!("List: {:?}", list);