ops.md 722 Bytes
Newer Older
1
2
3
4
## CUDA ops

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

5
- ActiveRotatedFilter
6
- AssignScoreWithK
7
- BallQuery
8
9
10
11
12
13
14
- BBoxOverlaps
- CARAFE
- CrissCrossAttention
- ContextBlock
- CornerPool
- Deformable Convolution v1/v2
- Deformable RoIPool
15
- DynamicScatter
16
- GatherPoints
17
18
- FurthestPointSample
- FurthestPointSampleWithDist
19
- GeneralizedAttention
20
- GroupPoints
21
- KNN
22
- MaskedConv
23
- MinAreaPolygon
24
- NMS
25
- PointsInPolygons
26
- PSAMask
27
- RiRoIAlignRotated
28
- RotatedFeatureAlign
29
- RoIPointPool3d
30
31
- RoIPool
- RoIAlign
32
- RoIAwarePool3d
33
34
35
36
37
- SimpleRoIAlign
- SigmoidFocalLoss
- SoftmaxFocalLoss
- SoftNMS
- Synchronized BatchNorm
38
- Voxelization
39
40
- ThreeInterpolate
- ThreeNN
41
- Weight standardization
42
- Correlation