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