Skip to content
/ Soft-NMS Public

Python and Pytorch two implements of Soft NMS algorithm

License

Notifications You must be signed in to change notification settings

DocF/Soft-NMS

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Python and Pytorch two implements of Soft NMS algorithm

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages