Skip to content

HsLOL/ExtensionOPs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Collection of C/C++/CUDA extensions.

  • NMS Op Extension (nms/ folder)
    • HBB\_NMS\_NMS_CPU/ folder
    • HBB\_NMS\_NMS_GPU/ folder
    • HBB\_Soft_NMS_CPU/ folder
    • OBB\_NMS_GPU/ folder
  • DCNv2 Op Extension (DCNv2/ folder)

Update A demo using C++ and CUDA to compile python extension.

  • demo (demo/ folder)

Reference:

https://github.com/ming71/CUDA
https://github.com/rbgirshick/py-faster-rcnn/tree/master/lib
https://github.com/bharatsingh430/soft-nms/tree/master/lib
https://github.com/tczhangzhi/pytorch-parallel
https://github.com/jinfagang/DCNv2_latest
Thanks for these awesome repos !:clap:
To be continue...