oa::vision::nms

Select score-ranked boxes with deterministic class-aware suppression. Equal scores select the smaller source index. NaN or infinite scores are ignored. All outputs remain device-resident and the operation never sorts or compacts on the host. Errors Returns an error unless boxes are nonempty FP32 [N, 4], scores are FP32 [N], classes are I32 [N], all inputs share an engine, thresholds are finite, iou_threshold is in [0, 1], and max_detections is positive.

FnRust public API

Function Documentation

pub fn nms( boxes: &Matrix, scores: &Matrix, classes: &Matrix, config: NmsConfig ) -> Result<NmsResult>
Select score-ranked boxes with deterministic class-aware suppression. Equal scores select the smaller source index. NaN or infinite scores are ignored. All outputs remain device-resident and the operation never sorts or compacts on the host. Errors Returns an error unless boxes are nonempty FP32 [N, 4], scores are FP32 [N], classes are I32 [N], all inputs share an engine, thresholds are finite, iou_threshold is in [0, 1], and max_detections is positive.