Open3D (C++ API)  0.18.0+252c867
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 
22 namespace open3d {
23 namespace ml {
24 namespace contrib {
25 
26 #ifdef BUILD_CUDA_MODULE
27 
34 std::vector<int64_t> NmsCUDAKernel(const float *boxes,
35  const float *scores,
36  int n,
37  double nms_overlap_thresh);
38 #endif
39 
46 std::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