CUDA opsΒΆ
We implement common CUDA ops used in detection, segmentation, etc.
ActiveRotatedFilter
AssignScoreWithK
BallQuery
BBoxOverlaps
CARAFE
CrissCrossAttention
ContextBlock
ConvexIoU
CornerPool
Deformable Convolution v1/v2
Deformable RoIPool
DiffIoURotated
DynamicScatter
GatherPoints
FurthestPointSample
FurthestPointSampleWithDist
GeneralizedAttention
GroupPoints
KNN
MaskedConv
MinAreaPolygon
NMS
NMS3D
PointsInPolygons
PSAMask
RiRoIAlignRotated
RotatedFeatureAlign
RoIPointPool3d
RoIPool
RoIAlign
RoIAwarePool3d
SimpleRoIAlign
SigmoidFocalLoss
SoftmaxFocalLoss
SoftNMS
Sparse Convolution
Synchronized BatchNorm
Voxelization
ThreeInterpolate
ThreeNN
Weight standardization
Correlation