Nms.h File Reference

(d32b4fc (Mon Sep 14 15:46:25 2026 -0700))

#include <algorithm>
#include <cstdint>
#include <numeric>
#include <vector>
#include "open3d/core/CUDAUtils.h"
#include "open3d/ml/contrib/IoUImpl.h"
#include "open3d/utility/Helper.h"

Go to the source code of this file.

Namespaces
namespace	open3d
namespace	open3d::ml
namespace	open3d::ml::contrib

Functions
std::vector< int64_t >	open3d::ml::contrib::NmsCPUKernel (const float *boxes, const float *scores, int n, double nms_overlap_thresh)
template<typename T >
std::vector< int64_t >	open3d::ml::contrib::SortIndexesDescending (const T *values, int64_t num)
OPEN3D_HOST_DEVICE int	open3d::ml::contrib::NmsGreedyKeepCore (const uint64_t *mask, const int64_t *sort_indices, int n, uint64_t *remv, int64_t *keep_indices)
std::vector< int64_t >	open3d::ml::contrib::NmsGreedyKeep (const uint64_t *mask, const int64_t *sort_indices, int n)