Skip to content

CanshangD/CUDA

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

codebase

Useful cuda codes with annotations inside .

RoIAlign

roi_align

I've heavily commented the code about RoIAlign for better usage . Two different version are included , and no much different effect founded between them.

rotate_roi_align

(Not checked yet!) RoIAlign for rotated rois . Not recommended before my checkout .

Rotated-NMS

Code for skew-bboxes NMS calculation.

DCNv2

Replaceable and faster implemention for conv.Directly apply to replace existed conv layer.
bug: not support apex mix precision training or half, in debuging.

ORN

Rotated conv layer.
Test demo also included for MNIST.(I've fixed a bug and made it supportable for pytorch with version higher than 1.0.0.)

About

useful cuda code .

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Cuda 64.1%
  • Python 26.3%
  • C++ 7.1%
  • C 2.4%
  • Shell 0.1%