2020-06-28 23:15:47 +08:00
|
|
|
## CUDA ops
|
|
|
|
|
|
|
|
We implement common CUDA ops used in detection, segmentation, etc.
|
|
|
|
|
2021-10-15 13:18:31 +08:00
|
|
|
- AssignScoreWithK
|
2021-09-23 21:08:00 +08:00
|
|
|
- BallQuery
|
2020-06-28 23:15:47 +08:00
|
|
|
- BBoxOverlaps
|
|
|
|
- CARAFE
|
|
|
|
- CrissCrossAttention
|
|
|
|
- ContextBlock
|
|
|
|
- CornerPool
|
|
|
|
- Deformable Convolution v1/v2
|
|
|
|
- Deformable RoIPool
|
2021-10-14 16:58:12 +08:00
|
|
|
- GatherPoints
|
2021-10-13 20:39:31 +08:00
|
|
|
- FurthestPointSample
|
|
|
|
- FurthestPointSampleWithDist
|
2020-06-28 23:15:47 +08:00
|
|
|
- GeneralizedAttention
|
2021-10-14 13:33:13 +08:00
|
|
|
- KNN
|
2020-06-28 23:15:47 +08:00
|
|
|
- MaskedConv
|
|
|
|
- NMS
|
|
|
|
- PSAMask
|
|
|
|
- RoIPool
|
|
|
|
- RoIAlign
|
|
|
|
- SimpleRoIAlign
|
|
|
|
- SigmoidFocalLoss
|
|
|
|
- SoftmaxFocalLoss
|
|
|
|
- SoftNMS
|
|
|
|
- Synchronized BatchNorm
|
2021-10-14 20:51:02 +08:00
|
|
|
- ThreeInterpolate
|
|
|
|
- ThreeNN
|
2020-06-28 23:15:47 +08:00
|
|
|
- Weight standardization
|
2021-09-23 21:08:41 +08:00
|
|
|
- Correlation
|