Skip to content

Latest commit

 

History

History
23 lines (21 loc) · 395 Bytes

ops.md

File metadata and controls

23 lines (21 loc) · 395 Bytes

CUDA ops

We implement common CUDA ops used in detection, segmentation, etc.

  • BBoxOverlaps
  • CARAFE
  • CrissCrossAttention
  • ContextBlock
  • CornerPool
  • Deformable Convolution v1/v2
  • Deformable RoIPool
  • GeneralizedAttention
  • MaskedConv
  • NMS
  • PSAMask
  • RoIPool
  • RoIAlign
  • SimpleRoIAlign
  • SigmoidFocalLoss
  • SoftmaxFocalLoss
  • SoftNMS
  • Synchronized BatchNorm
  • Weight standardization