ops.md 780 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
- BBoxOverlaps
- CARAFE
- CrissCrossAttention
- ContextBlock
12
- ConvexIoU
13
14
15
- CornerPool
- Deformable Convolution v1/v2
- Deformable RoIPool
16
- DiffIoURotated
17
- DynamicScatter
18
- GatherPoints
19
20
- FurthestPointSample
- FurthestPointSampleWithDist
21
- GeneralizedAttention
22
- GroupPoints
23
- KNN
24
- MaskedConv
25
- MinAreaPolygon
26
- NMS
27
- NMS3D
28
- PointsInPolygons
29
- PSAMask
30
- RiRoIAlignRotated
31
- RotatedFeatureAlign
32
- RoIPointPool3d
33
34
- RoIPool
- RoIAlign
35
- RoIAwarePool3d
36
37
38
39
- SimpleRoIAlign
- SigmoidFocalLoss
- SoftmaxFocalLoss
- SoftNMS
40
- Sparse Convolution
41
- Synchronized BatchNorm
42
- Voxelization
43
44
- ThreeInterpolate
- ThreeNN
45
- Weight standardization
46
- Correlation