Open3D (C++ API)  0.18.0+252c867
IoU.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 #pragma once
9 
10 namespace open3d {
11 namespace ml {
12 namespace contrib {
13 
14 #ifdef BUILD_CUDA_MODULE
21 void IoUBevCUDAKernel(const float *boxes_a,
22  const float *boxes_b,
23  float *iou,
24  int num_a,
25  int num_b);
26 
32 void IoU3dCUDAKernel(const float *boxes_a,
33  const float *boxes_b,
34  float *iou,
35  int num_a,
36  int num_b);
37 
38 #endif
39 
46 void IoUBevCPUKernel(const float *boxes_a,
47  const float *boxes_b,
48  float *iou,
49  int num_a,
50  int num_b);
51 
57 void IoU3dCPUKernel(const float *boxes_a,
58  const float *boxes_b,
59  float *iou,
60  int num_a,
61  int num_b);
62 
63 } // namespace contrib
64 } // namespace ml
65 } // namespace open3d
void IoUBevCPUKernel(const float *boxes_a, const float *boxes_b, float *iou, int num_a, int num_b)
Definition: IoU.cpp:18
void IoU3dCPUKernel(const float *boxes_a, const float *boxes_b, float *iou, int num_a, int num_b)
Definition: IoU.cpp:33
Definition: PinholeCameraIntrinsic.cpp:16