megengine.functional.nn.nms¶
- nms(boxes, scores, iou_thresh, max_output=None)[源代码]¶
根据小方框的 IoU 值进行 非最大值抑制(NMS)。
- 参数
- 返回类型
- 返回
按scores排序的NMS保留的元素的指数。
注解
应该保留max_output并且跟踪有效正值。
实际案例
import numpy as np from megengine import tensor import megengine.functional as F x = np.zeros((100,4)) np.random.seed(42) x[:,:2] = np.random.rand(100,2)*20 x[:,2:] = np.random.rand(100,2)*20 + 100 scores = tensor(np.random.rand(100)) inp = tensor(x) result = F.vision.nms(inp, scores, iou_thresh=0.7) print(result.numpy())
输出:
[75 69]