haar-cascade-nms

Extend opencv haar-cascade detector to filter detections with Non-Maxima Suppression (NMS)


Keywords
object-recognition, object-detection, object-localization, cascade-detectors, haar-cascade, opencv
License
MIT
Install
pip install haar-cascade-nms==1.1.1

Documentation

OpenCV Haar cascade with Non-Maxima Suppression

This simple project provides a Detector class extending the opencv CascadeDetectorclass with a new function: detectAndFilter.
This function performs the detection using the cascade and filter the detections based on the score and their overlap to yield the best detections that do not overlap above the selected overlap-threshold.

For more details about Non-Maxima Suppression, you can refer to my article about Multi-Template Matching, which uses a similar strategy (or other online resources).
Here the NMS shipped with OpenCV is used.

Installation

Using pip:
pip install haar-cascade-nms

Then to use in your scripts:

from haarcascade import Detector
myDetector = Detector(filepathToCascade)
myDetector.detectAndFilter(image, etc...)

See the tutorial notebook for a more detailed documentation.