From 857be2577cc18cefc8386cab7d52e0b06efe5f17 Mon Sep 17 00:00:00 2001 From: JordanCheney Date: Mon, 29 Sep 2014 12:20:09 -0400 Subject: [PATCH] Exposed minNeighbors parameter in Viola Jones detector --- openbr/plugins/cascade.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/openbr/plugins/cascade.cpp b/openbr/plugins/cascade.cpp index 252778e..bd13f7f 100644 --- a/openbr/plugins/cascade.cpp +++ b/openbr/plugins/cascade.cpp @@ -199,6 +199,7 @@ class CascadeTransform : public MetaTransform Q_OBJECT Q_PROPERTY(QString model READ get_model WRITE set_model RESET reset_model STORED false) Q_PROPERTY(int minSize READ get_minSize WRITE set_minSize RESET reset_minSize STORED false) + Q_PROPERTY(int minNeighbors READ get_minNeighbors WRITE set_minNeighbors RESET reset_minNeighbors STORED false) Q_PROPERTY(bool ROCMode READ get_ROCMode WRITE set_ROCMode RESET reset_ROCMode STORED false) // Training parameters @@ -223,6 +224,7 @@ class CascadeTransform : public MetaTransform BR_PROPERTY(QString, model, "FrontalFace") BR_PROPERTY(int, minSize, 64) + BR_PROPERTY(int, minNeighbors, 5) BR_PROPERTY(bool, ROCMode, false) // Training parameters - Default values provided trigger OpenCV defaults @@ -392,8 +394,8 @@ class CascadeTransform : public MetaTransform std::vector rects; std::vector rejectLevels; std::vector levelWeights; - if (ROCMode) cascade->detectMultiScale(m, rects, rejectLevels, levelWeights, 1.2, 5, (enrollAll ? 0 : CASCADE_FIND_BIGGEST_OBJECT) | CASCADE_SCALE_IMAGE, Size(minSize, minSize), Size(), true); - else cascade->detectMultiScale(m, rects, 1.2, 5, enrollAll ? 0 : CASCADE_FIND_BIGGEST_OBJECT, Size(minSize, minSize)); + if (ROCMode) cascade->detectMultiScale(m, rects, rejectLevels, levelWeights, 1.2, minNeighbors, (enrollAll ? 0 : CASCADE_FIND_BIGGEST_OBJECT) | CASCADE_SCALE_IMAGE, Size(minSize, minSize), Size(), true); + else cascade->detectMultiScale(m, rects, 1.2, minNeighbors, enrollAll ? 0 : CASCADE_FIND_BIGGEST_OBJECT, Size(minSize, minSize)); if (!enrollAll && rects.empty()) rects.push_back(Rect(0, 0, m.cols, m.rows)); -- libgit2 0.21.4