dingchang 63a6cbe913
[Feature] Add fps op from mmdet3d (#1337)
* add ops (furthest point sample) in mmdet3d

* refactor code

* refactor code

* fix typo

* fix typo

* fix typo

* refactor code

* fix typo

* define DIVUP in common_cuda_helper.hpp
2021-10-13 20:39:31 +08:00

473 B

CUDA ops

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

  • BallQuery
  • BBoxOverlaps
  • CARAFE
  • CrissCrossAttention
  • ContextBlock
  • CornerPool
  • Deformable Convolution v1/v2
  • Deformable RoIPool
  • FurthestPointSample
  • FurthestPointSampleWithDist
  • GeneralizedAttention
  • MaskedConv
  • NMS
  • PSAMask
  • RoIPool
  • RoIAlign
  • SimpleRoIAlign
  • SigmoidFocalLoss
  • SoftmaxFocalLoss
  • SoftNMS
  • Synchronized BatchNorm
  • Weight standardization
  • Correlation