Skip to content

merisgong208/LibTorch_NMS

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LibTorch_NMS

Why I do this?

when I write A Libtorch API for detector. I find all NMS in GitHub are access to python version. There is not The Api for Only C++ . So I do for every one when need NMS_cuda in your C++ API

How to Work

bash make.sh
we get libnms.so

Test

we give a easily example to test
g++ -o nms nms.cpp -L. -lnms && ./nms

Be Careful

when U use this Api, REMEBER the Det U must be sorting scores according to descending order ;.

About

NMS_API For C++ API in pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Cuda 79.8%
  • Shell 10.5%
  • C++ 9.7%