iou3d_nms.h 1.3 KB

123456789101112131415161718192021222324252627282930313233
  1. // Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. #ifndef IOU3D_NMS_H
  15. #define IOU3D_NMS_H
  16. #include <cuda.h>
  17. #include <cuda_runtime_api.h>
  18. #include <paddle/extension.h>
  19. #include <vector>
  20. std::vector<paddle::Tensor> boxes_overlap_bev_gpu(
  21. const paddle::Tensor &boxes_a, const paddle::Tensor &boxes_b);
  22. std::vector<paddle::Tensor> boxes_iou_bev_gpu(
  23. const paddle::Tensor &boxes_a_tensor, const paddle::Tensor &boxes_b_tensor);
  24. std::vector<paddle::Tensor> nms_gpu(const paddle::Tensor &boxes,
  25. float nms_overlap_thresh);
  26. std::vector<paddle::Tensor> nms_normal_gpu(const paddle::Tensor &boxes,
  27. float nms_overlap_thresh);
  28. #endif