ops.md 783 Bytes
Newer Older
Zaida Zhou's avatar
Zaida Zhou committed
1
2
## CUDA 算子

3
4
MMCV 提供了检测、分割等任务中常用的 CUDA 算子

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
- KNN
24
- MaskedConv
25
- MinAreaPolygon
26
- NMS
27
- NMS3D
28
- PointsInPolygons
29
- PSAMask
30
- RotatedFeatureAlign
31
- RoIPointPool3d
32
- RoIPool
33
- RiRoIAlignRotated
34
- 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