".github/vscode:/vscode.git/clone" did not exist on "401289fa99ef2ea9e3b43e59ab1602882e63dc60"
ops.md 740 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
14
15
- CornerPool
- Deformable Convolution v1/v2
- Deformable RoIPool
16
- DynamicScatter
17
- GatherPoints
18
19
- FurthestPointSample
- FurthestPointSampleWithDist
20
- GeneralizedAttention
21
- KNN
22
- MaskedConv
23
- MinAreaPolygon
24
- NMS
25
- PointsInPolygons
26
- PSAMask
27
- RotatedFeatureAlign
28
- RoIPointPool3d
29
- RoIPool
30
- RiRoIAlignRotated
31
- RoIAlign
32
- RoIAwarePool3d
33
34
35
36
- SimpleRoIAlign
- SigmoidFocalLoss
- SoftmaxFocalLoss
- SoftNMS
37
- Sparse Convolution
38
- Synchronized BatchNorm
39
- Voxelization
40
41
- ThreeInterpolate
- ThreeNN
42
- Weight standardization
43
- Correlation