oa::FnDetection::nms

Deterministic class-aware NMS. boxes are FP32 [N,4] cx/cy/w/h, scores FP32 [N], and classes Int32 [N]. The implementation is GPU-resident and records as one graph node; it never sorts or compacts on the host.

Function Documentation

NmsResult oa::FnDetection::nms( const Matrix & inBoxes, const Matrix & inScores, const Matrix & inClasses, const NmsConfig & inConfig = {} )
Deterministic class-aware NMS. boxes are FP32 [N,4] cx/cy/w/h, scores FP32 [N], and classes Int32 [N]. The implementation is GPU-resident and records as one graph node; it never sorts or compacts on the host.

Parameters

inBoxes
const Matrix &

inScores
const Matrix &

inClasses
const Matrix &

inConfig
const NmsConfig &

Default: {}

Returns

NmsResult

The declared return value.