Soft-NMS
Implements of Soft NMS algorithm
Original PaperοΌImproving Object Detection With One Line of Code
Description
This repo include :
- Python version of Soft NMS algorithm
- Pytorch version of Soft NMS algorithm
Speed analysis
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 [email protected] |
| pytorch cpu version | 21.778 | i9-7900X [email protected] |
| pytorch gpu version | 60.128 | TITAN Xp |
ToDo
- PyTorch CUDA version
- speed analysis
- Keras version