mmcv/ops.md at 5c25ae1a19a22d3ae3ec844d4d6ebbb7af3fe091 - mmcv - Gitea: Git with a cup of tea

mirrors/mmcv

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

dingchang 5c25ae1a19

[Feature] Add roipoint_pool3d op from mmdet3d (#1358 )

* add ops (roipoint_pool3d) in mmdet3d

* refactor code

* fix typo

* add unit test

* refactor code

* refactor code

* refactor code

* fix typo

2021-10-15 17:15:53 +08:00

559 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
GatherPoints
FurthestPointSample
FurthestPointSampleWithDist
GeneralizedAttention
KNN
MaskedConv
NMS
PSAMask
RoIPointPool3d
RoIPool
RoIAlign
SimpleRoIAlign
SigmoidFocalLoss
SoftmaxFocalLoss
SoftNMS
Synchronized BatchNorm
ThreeInterpolate
ThreeNN
Weight standardization
Correlation