Fix NMS in yolo detector

nicholaskajoh / ivy

Video-based object counting software.

MIT License

428 stars 170 forks source link

Fix NMS in yolo detector #44

Closed JonathanSamelson closed 4 years ago

JonathanSamelson commented 4 years ago

Hi, Thank you very much for your work!

I discovered a bug while working with your project. If you don't update the list of classes and the list of confidences, they won't match the boxes if there was a box deleted by the NMS.

nicholaskajoh commented 4 years ago

Haha! Nice. Thanks @JonathanSamelson. Merged!!!