crazyvertigo/soft-nms

Python and Pytorch two implements of Soft NMS algorithm

Soft-NMS

Implements of Soft NMS algorithm

Original Paper：Improving Object Detection With One Line of Code

Description

This repo include :

1. Python version of Soft NMS algorithm
2. 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