feat: Added a manual implementation of nms

This commit is contained in:
uttarayan21
2025-08-07 15:45:54 +05:30
parent 2c43f657aa
commit e91ae5b865
5 changed files with 46 additions and 80 deletions

View File

@@ -27,7 +27,9 @@ pub fn main() -> Result<()> {
let output = model
.detect_faces(
array.clone(),
FaceDetectionConfig::default().with_threshold(detect.threshold),
FaceDetectionConfig::default()
.with_threshold(detect.threshold)
.with_nms_threshold(detect.nms_threshold),
)
.change_context(errors::Error)
.attach_printable("Failed to detect faces")?;