Implements of Soft NMS algorithm
Original Paper：Improving Object Detection With One Line of Code
This repo include :
- Python version of Soft NMS algorithm
- Pytorch version of Soft NMS algorithm
The average running time was obtained using 1000 tests, and 100 candidate boxes and scores were randomly generated for each test. Because the algorithm has many complex operations that are not suitable for GPU processing, the GPU version of the algorithm is rather slow.
| version | run time(ms) | device |
|---|---|---|
| python cpu version | 21.472 | i9-7900X CPU@3.3GHz |
| pytorch cpu version | 21.778 | i9-7900X CPU@3.3GHz |
| pytorch gpu version | 60.128 | TITAN Xp |
- PyTorch CUDA version
- speed analysis
- Keras version