Open3D (C++ API)  0.18.0
Loading...
Searching...
No Matches
Nms.h
Go to the documentation of this file.
1// ----------------------------------------------------------------------------
2// - Open3D: www.open3d.org -
3// ----------------------------------------------------------------------------
4// Copyright (c) 2018-2023 www.open3d.org
5// SPDX-License-Identifier: MIT
6// ----------------------------------------------------------------------------
7//
8// Reference:
9// https://github.com/open-mmlab/OpenPCDet/blob/master/pcdet/ops/iou3d_nms/src/iou3d_nms_kernel.cu
10//
11// Reference:
12// https://github.com/open-mmlab/mmdetection3d/blob/master/mmdet3d/ops/iou3d/src/iou3d_kernel.cu
13// 3D IoU Calculation and Rotated NMS(modified from 2D NMS written by others)
14// Written by Shaoshuai Shi
15// All Rights Reserved 2019-2020.
16
17#pragma once
18
19#include <cstdint>
20#include <vector>
21
22namespace open3d {
23namespace ml {
24namespace contrib {
25
26#ifdef BUILD_CUDA_MODULE
27
34std::vector<int64_t> NmsCUDAKernel(const float *boxes,
35 const float *scores,
36 int n,
37 double nms_overlap_thresh);
38#endif
39
46std::vector<int64_t> NmsCPUKernel(const float *boxes,
47 const float *scores,
48 int n,
49 double nms_overlap_thresh);
50
51} // namespace contrib
52} // namespace ml
53} // namespace open3d
std::vector< int64_t > NmsCPUKernel(const float *boxes, const float *scores, int n, double nms_overlap_thresh)
Definition Nms.cpp:120
Definition PinholeCameraIntrinsic.cpp:16