Open3D (C++ API)  0.17.0
RoiPoolKernel.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//
9// Based on PointRCNN Library (MIT License):
10// https://github.com/sshaoshuai/PointRCNN
11//
12// Copyright (c) 2019 Shaoshuai Shi
13//
14// Permission is hereby granted, free of charge, to any person obtaining a
15// copy of this software and associated documentation files (the "Software"),
16// to deal in the Software without restriction, including without limitation
17// the rights to use, copy, modify, merge, publish, distribute, sublicense,
18// and/or sell copies of the Software, and to permit persons to whom the
19// Software is furnished to do so, subject to the following conditions:
20//
21// The above copyright notice and this permission notice shall be included in
22// all copies or substantial portions of the Software.
23//
24// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
25// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
26// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
27// THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
28// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
29// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
30// DEALINGS IN THE SOFTWARE.
31//
32//***************************************************************************************/
33
34#pragma once
35
36namespace open3d {
37namespace ml {
38namespace contrib {
39
40#ifdef BUILD_CUDA_MODULE
41
42void roipool3dLauncher(int batch_size,
43 int pts_num,
44 int boxes_num,
45 int feature_in_len,
46 int sampled_pts_num,
47 const float *xyz,
48 const float *boxes3d,
49 const float *pts_feature,
50 float *pooled_features,
51 int *pooled_empty_flag);
52#endif
53
54} // namespace contrib
55} // namespace ml
56} // namespace open3d
Definition: PinholeCameraIntrinsic.cpp:16