Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
hukaixuan19970627
GitHub Repository: hukaixuan19970627/yolov5_obb
 Path: blob/master/utils/nms_rotated/src/poly_nms_cpu.cpp
1601 views
1
#include <torch/extension.h>

2


3
template <typename scalar_t>

4
at::Tensor poly_nms_cpu_kernel(const at::Tensor& dets, const float threshold) {

5


6


7