├── .gitignore ├── common ├── include │ ├── cpu_nms.hpp │ └── gpu_nms.hpp ├── main.cpp ├── make.sh └── src │ ├── cpu_nms.cc │ └── nms_kernel.cu ├── libtorch └── nms │ ├── cpu │ ├── nms_cpu.cpp │ ├── nms_cpu.h │ └── nms_cpu.hpp │ ├── cuda │ ├── cuda_helpers.h │ ├── nms_cuda.cu │ ├── nms_cuda.h │ └── nms_cuda.hpp │ └── nms.h └── readme.md /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/.gitignore -------------------------------------------------------------------------------- /common/include/cpu_nms.hpp: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /common/include/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/common/include/gpu_nms.hpp -------------------------------------------------------------------------------- /common/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/common/main.cpp -------------------------------------------------------------------------------- /common/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/common/make.sh -------------------------------------------------------------------------------- /common/src/cpu_nms.cc: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /common/src/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/common/src/nms_kernel.cu -------------------------------------------------------------------------------- /libtorch/nms/cpu/nms_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/libtorch/nms/cpu/nms_cpu.cpp -------------------------------------------------------------------------------- /libtorch/nms/cpu/nms_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/libtorch/nms/cpu/nms_cpu.h -------------------------------------------------------------------------------- /libtorch/nms/cpu/nms_cpu.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/libtorch/nms/cpu/nms_cpu.hpp -------------------------------------------------------------------------------- /libtorch/nms/cuda/cuda_helpers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/libtorch/nms/cuda/cuda_helpers.h -------------------------------------------------------------------------------- /libtorch/nms/cuda/nms_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/libtorch/nms/cuda/nms_cuda.cu -------------------------------------------------------------------------------- /libtorch/nms/cuda/nms_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/libtorch/nms/cuda/nms_cuda.h -------------------------------------------------------------------------------- /libtorch/nms/cuda/nms_cuda.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/libtorch/nms/cuda/nms_cuda.hpp -------------------------------------------------------------------------------- /libtorch/nms/nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/libtorch/nms/nms.h -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasjinreal/libnms/HEAD/readme.md --------------------------------------------------------------------------------