Open3D (C++ API)  0.18.0
Loading...
Searching...
No Matches
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
10namespace open3d {
11namespace ml {
12namespace contrib {
13
14#ifdef BUILD_CUDA_MODULE
21void IoUBevCUDAKernel(const float *boxes_a,
22 const float *boxes_b,
23 float *iou,
24 int num_a,
25 int num_b);
26
32void 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
46void IoUBevCPUKernel(const float *boxes_a,
47 const float *boxes_b,
48 float *iou,
49 int num_a,
50 int num_b);
51
57void 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