HsLOL / ExtensionOPs

Some C++/C/CUDA Extension

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Some C++/C/CUDA Extension


Languages

Language:Cuda 33.8%Language:C++ 31.5%Language:Python 24.8%Language:C 5.3%Language:Cython 4.4%Language:Makefile 0.1%Language:Shell 0.0%