ops.md 772 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
- PointsInPolygons
28
- PSAMask
29
- RiRoIAlignRotated
30
- RotatedFeatureAlign
31
- RoIPointPool3d
32
33
- RoIPool
- RoIAlign
34
- RoIAwarePool3d
35
36
37
38
- SimpleRoIAlign
- SigmoidFocalLoss
- SoftmaxFocalLoss
- SoftNMS
39
- Sparse Convolution
40
- Synchronized BatchNorm
41
- Voxelization
42
43
- ThreeInterpolate
- ThreeNN
44
- Weight standardization
45
- Correlation