Home

Awesome

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.

versionrun time(ms)device
python cpu version21.472i9-7900X CPU@3.3GHz
pytorch cpu version21.778i9-7900X CPU@3.3GHz
pytorch gpu version60.128TITAN Xp

ToDo