2022-06-15 10:57:29 +08:00
|
|
|
## ops
|
2020-06-28 23:15:47 +08:00
|
|
|
|
2022-06-15 10:57:29 +08:00
|
|
|
We implement common ops used in detection, segmentation, etc.
|
2020-06-28 23:15:47 +08:00
|
|
|
|
2022-07-22 19:30:01 +08:00
|
|
|
| Device | CPU | CUDA | MLU | MPS |
|
|
|
|
| ---------------------------- | --- | ---- | --- | --- |
|
|
|
|
| ActiveRotatedFilter | √ | √ | | |
|
|
|
|
| AssignScoreWithK | | √ | | |
|
|
|
|
| BallQuery | | √ | | |
|
|
|
|
| BBoxOverlaps | | √ | √ | √ |
|
|
|
|
| BorderAlign | | √ | | |
|
|
|
|
| BoxIouRotated | √ | √ | | |
|
2022-10-13 17:26:14 +08:00
|
|
|
| BoxIouQuadri | √ | √ | | |
|
2022-08-19 10:36:26 +08:00
|
|
|
| CARAFE | | √ | √ | |
|
2022-07-22 19:30:01 +08:00
|
|
|
| ChamferDistance | | √ | | |
|
|
|
|
| CrissCrossAttention | | √ | | |
|
|
|
|
| ContourExpand | √ | | | |
|
|
|
|
| ConvexIoU | | √ | | |
|
|
|
|
| CornerPool | | √ | | |
|
|
|
|
| Correlation | | √ | | |
|
|
|
|
| Deformable Convolution v1/v2 | √ | √ | | |
|
2022-08-28 22:37:28 +08:00
|
|
|
| Deformable RoIPool | | √ | √ | |
|
2022-07-22 19:30:01 +08:00
|
|
|
| DiffIoURotated | | √ | | |
|
|
|
|
| DynamicScatter | | √ | | |
|
|
|
|
| FurthestPointSample | | √ | | |
|
|
|
|
| FurthestPointSampleWithDist | | √ | | |
|
|
|
|
| FusedBiasLeakyrelu | | √ | | |
|
|
|
|
| GatherPoints | | √ | | |
|
|
|
|
| GroupPoints | | √ | | |
|
2022-10-24 15:58:58 +08:00
|
|
|
| Iou3d | | √ | √ | |
|
2022-07-22 19:30:01 +08:00
|
|
|
| KNN | | √ | | |
|
2022-08-21 23:21:54 +08:00
|
|
|
| MaskedConv | | √ | √ | |
|
2022-07-22 19:30:01 +08:00
|
|
|
| MergeCells | | √ | | |
|
|
|
|
| MinAreaPolygon | | √ | | |
|
|
|
|
| ModulatedDeformConv2d | √ | √ | | |
|
|
|
|
| MultiScaleDeformableAttn | | √ | | |
|
|
|
|
| NMS | √ | √ | √ | |
|
|
|
|
| NMSRotated | √ | √ | | |
|
2022-10-13 17:26:14 +08:00
|
|
|
| NMSQuadri | √ | √ | | |
|
2022-07-22 19:30:01 +08:00
|
|
|
| PixelGroup | √ | | | |
|
|
|
|
| PointsInBoxes | √ | √ | | |
|
|
|
|
| PointsInPolygons | | √ | | |
|
|
|
|
| PSAMask | √ | √ | √ | |
|
|
|
|
| RotatedFeatureAlign | √ | √ | | |
|
2022-09-08 17:19:12 +08:00
|
|
|
| RoIPointPool3d | | √ | √ | |
|
2022-07-22 19:30:01 +08:00
|
|
|
| RoIPool | | √ | √ | |
|
|
|
|
| RoIAlignRotated | √ | √ | √ | |
|
|
|
|
| RiRoIAlignRotated | | √ | | |
|
|
|
|
| RoIAlign | √ | √ | √ | |
|
2022-11-03 11:13:34 +08:00
|
|
|
| RoIAwarePool3d | | √ | √ | |
|
2022-07-22 19:30:01 +08:00
|
|
|
| SAConv2d | | √ | | |
|
|
|
|
| SigmoidFocalLoss | | √ | √ | |
|
|
|
|
| SoftmaxFocalLoss | | √ | | |
|
|
|
|
| SoftNMS | | √ | | |
|
|
|
|
| Sparse Convolution | | √ | | |
|
|
|
|
| Synchronized BatchNorm | | √ | | |
|
|
|
|
| ThreeInterpolate | | √ | | |
|
2022-08-23 15:18:47 +08:00
|
|
|
| ThreeNN | | √ | √ | |
|
2022-07-22 19:30:01 +08:00
|
|
|
| TINShift | | √ | √ | |
|
|
|
|
| UpFirDn2d | | √ | | |
|
|
|
|
| Voxelization | √ | √ | | |
|
2022-07-23 23:08:32 +08:00
|
|
|
| PrRoIPool | | √ | | |
|