/TorchEx

Collection of PyTorch customized operators.

Primary LanguageCuda

TorchEx

Collection of PyTorch customized operators for personal projects.

Supported GPU Operators:

  • Weighted NMS used in RangeDet.
  • IngroupIndices used in SST.
  • DynamicPointPooling (Group Correction) used in FSD.
  • Connected Component Labeling used in FSD.

Usage

With PyTorch installation, simply run pip install -e ..