CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutSign UpSign In
hukaixuan19970627

Real-time collaboration for Jupyter Notebooks, Linux Terminals, LaTeX, VS Code, R IDE, and more,
all in one place. Commercial Alternative to JupyterHub.

GitHub Repository: hukaixuan19970627/yolov5_obb
Path: blob/master/utils/nms_rotated/src/poly_nms_cpu.cpp
Views: 475
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