kreuzberg 4.5.0

High-performance document intelligence library for Rust. Extract text, metadata, and structured data from PDFs, Office documents, images, and 88+ formats with async/sync APIs.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use crate::layout::types::LayoutDetection;

/// Standard greedy Non-Maximum Suppression.
///
/// Sorts detections by confidence (descending), then iteratively removes
/// detections that have IoU > `iou_threshold` with any higher-confidence detection.
///
/// This is required for YOLO models. RT-DETR is NMS-free.
pub fn greedy_nms(detections: &mut Vec<LayoutDetection>, iou_threshold: f32) {
    LayoutDetection::sort_by_confidence_desc(detections);

    let n = detections.len();
    let mut keep = vec![true; n];

    for i in 0..n {
        if !keep[i] {
            continue;
        }
        for j in (i + 1)..n {
            if !keep[j] {
                continue;
            }
            if detections[i].bbox.iou(&detections[j].bbox) > iou_threshold {
                keep[j] = false;
            }
        }
    }

    let mut idx = 0;
    detections.retain(|_| {
        let k = keep[idx];
        idx += 1;
        k
    });
}