cn.smartjavaai.common.utils

类 NMSUtils

java.lang.Object

cn.smartjavaai.common.utils.NMSUtils

public class NMSUtils
extends Object

作者:

dwj

构造器概要

构造器

构造器和说明
NMSUtils()

方法概要

限定符和类型	方法和说明
static ai.djl.ndarray.NDArray	batchedNms(ai.djl.ndarray.NDArray boxes, ai.djl.ndarray.NDArray scores, ai.djl.ndarray.NDArray idxs, float iouThreshold, ai.djl.ndarray.NDManager manager) 批量执行 NMS，输入 NDArray 形式的 boxes、scores 和 idxs，返回保留的索引列表
static int[]	mtcnnNms(ai.djl.ndarray.NDArray boxes, ai.djl.ndarray.NDArray scores, float iouThreshold)
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 阈值，超过该阈值则认为有重叠

返回:

保留框的索引列表

batchedNms

public static ai.djl.ndarray.NDArray batchedNms(ai.djl.ndarray.NDArray boxes,
                                                ai.djl.ndarray.NDArray scores,
                                                ai.djl.ndarray.NDArray idxs,
                                                float iouThreshold,
                                                ai.djl.ndarray.NDManager manager)

批量执行 NMS，输入 NDArray 形式的 boxes、scores 和 idxs，返回保留的索引列表

参数:

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

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

idxs - NDArray 形状为 (N,)，每个 box 对应的 batch id

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

返回:

批量保留框的索引列表

mtcnnNms

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