/Soft-NMS

Python and Pytorch two implements of Soft NMS algorithm

Primary LanguagePythonMIT LicenseMIT

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 CPU@3.3GHz
pytorch cpu version 21.778 i9-7900X CPU@3.3GHz
pytorch gpu version 60.128 TITAN Xp

ToDo

  • PyTorch CUDA version
  • speed analysis
  • Keras version