Skip to content

Latest commit

 

History

History
17 lines (17 loc) · 924 Bytes

README.md

File metadata and controls

17 lines (17 loc) · 924 Bytes

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...