reusex::vision::torchvision Namespace Reference

Functions
at::Tensor	nms_kernel (const at::Tensor &dets, const at::Tensor &scores, double iou_threshold)
	torchvision's reference CPU non-maximum-suppression kernel.

Function Documentation

nms_kernel()

at::Tensor reusex::vision::torchvision::nms_kernel	(	const at::Tensor &	dets,
		const at::Tensor &	scores,
		double	iou_threshold )

torchvision's reference CPU non-maximum-suppression kernel.

Parameters

dets	Boxes [N, 4] in (x1, y1, x2, y2) order, CPU, floating point.
scores	Confidence scores [N], same dtype and device as dets.
iou_threshold	Boxes are suppressed when IoU is strictly greater than this value.

Returns

int64 tensor of kept indices, in descending score order.

Throws c10::Error (via TORCH_CHECK) if the tensors are not on the CPU, have mismatched dtypes, or have the wrong rank/extent.