Shortcuts

ops

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

Device CPU CUDA MLU MPS
ActiveRotatedFilter √ √
AssignScoreWithK √
BallQuery √
BBoxOverlaps √ √ √
BorderAlign √
BoxIouRotated √ √
BoxIouQuadri √ √
CARAFE √ √
ChamferDistance √
CrissCrossAttention √
ContourExpand √
ConvexIoU √
CornerPool √
Correlation √
Deformable Convolution v1/v2 √ √
Deformable RoIPool √ √
DiffIoURotated √
DynamicScatter √
FurthestPointSample √
FurthestPointSampleWithDist √
FusedBiasLeakyrelu √
GatherPoints √
GroupPoints √
Iou3d √ √
KNN √
MaskedConv √ √
MergeCells √
MinAreaPolygon √
ModulatedDeformConv2d √ √
MultiScaleDeformableAttn √
NMS √ √ √
NMSRotated √ √
NMSQuadri √ √
PixelGroup √
PointsInBoxes √ √
PointsInPolygons √
PSAMask √ √ √
RotatedFeatureAlign √ √
RoIPointPool3d √ √
RoIPool √ √
RoIAlignRotated √ √ √
RiRoIAlignRotated √
RoIAlign √ √ √
RoIAwarePool3d √
SAConv2d √
SigmoidFocalLoss √ √
SoftmaxFocalLoss √
SoftNMS √
Sparse Convolution √
Synchronized BatchNorm √
ThreeInterpolate √
ThreeNN √ √
TINShift √ √
UpFirDn2d √
Voxelization √ √
PrRoIPool √