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