mmcv/ops.md at d4d6ea09e35f1b8800d268d97ce29c7453cae3cc

mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00

[Feature] Add group points ops from mmdet3d (#1415 )

* add op (group points) and its related ops (ball query and knn) in mmdet3d

* refactor code

* fix typo

* refactor code

* fix typo

* refactor code

* make input contiguous

Co-authored-by: zhouzaida <zhouzaida@163.com>

2021-10-23 14:01:31 +08:00

622 B

Raw Blame History

CUDA ops

We implement common CUDA ops used in detection, segmentation, etc.

AssignScoreWithK
BallQuery
BBoxOverlaps
CARAFE
CrissCrossAttention
ContextBlock
CornerPool
Deformable Convolution v1/v2
Deformable RoIPool
DynamicScatter
GatherPoints
FurthestPointSample
FurthestPointSampleWithDist
GeneralizedAttention
GroupPoints
KNN
MaskedConv
NMS
PSAMask
RoIPointPool3d
RoIPool
RoIAlign
RoIAwarePool3d
SimpleRoIAlign
SigmoidFocalLoss
SoftmaxFocalLoss
SoftNMS
Synchronized BatchNorm
Voxelization
ThreeInterpolate
ThreeNN
Weight standardization
Correlation

622 B Raw Blame History

CUDA ops

622 B

Raw Blame History