rykov8 / ssd_keras

Port of Single Shot MultiBox Detector to Keras
MIT License
1.1k stars 553 forks source link

Why does BBoxUtility.detection_out() set confidence_threshold to 0.01 by default? #89

Open neil454 opened 7 years ago

neil454 commented 7 years ago

In fact, in videotest.py, VideoTest.run() has a parameter conf_thresh, which is set to the expected value of 0.6, but then when calling detection_out(), it doesn't pass conf_thresh at all. Instead it just thresholds after the fact.

I've found that I can get substantial FPS improvements by passing the conf_thresh of 0.6 into detection_out() (usually around 1.5x increase)

EDIT: Actually that 1.5x increase is only for the default COCO model. For my own re-trained models on different datasets, I've gone from 2 FPS to around 25 FPS when using this fix.