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 ..