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