cn.smartjavaai.common.utils

类 NMSUtils

java.lang.Object

cn.smartjavaai.common.utils.NMSUtils

public class NMSUtils
extends Object

作者:

dwj

构造器概要

构造器

构造器和说明
NMSUtils()

方法概要

限定符和类型	方法和说明
static int[]	nms(ai.djl.ndarray.NDArray boxes, ai.djl.ndarray.NDArray scores, float iouThreshold) 通用 NMS 方法，输入 NDArray 形式的 boxes 和 scores，返回保留的索引列表

从类继承的方法 java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

构造器详细资料

NMSUtils

public NMSUtils()

方法详细资料

nms

public static int[] nms(ai.djl.ndarray.NDArray boxes,
                        ai.djl.ndarray.NDArray scores,
                        float iouThreshold)

通用 NMS 方法，输入 NDArray 形式的 boxes 和 scores，返回保留的索引列表

参数:

boxes - NDArray 形状为 (N, 4)，格式为 [x1, y1, x2, y2]

scores - NDArray 形状为 (N,) 或 (N,1)，每个 box 的置信度

iouThreshold - IOU 阈值，超过该阈值则认为有重叠

返回:

保留框的索引列表