diff --git a/openbr/plugins/distance/unit.cpp b/openbr/plugins/distance/unit.cpp index 491ffad..e2d60ed 100644 --- a/openbr/plugins/distance/unit.cpp +++ b/openbr/plugins/distance/unit.cpp @@ -76,12 +76,18 @@ class UnitDistance : public Distance float compare(const Template &target, const Template &query) const { - return a * (distance->compare(target, query) - b); + return normalize(distance->compare(target, query)); } float compare(const cv::Mat &target, const cv::Mat &query) const { - return a * (distance->compare(target, query) - b); + return normalize(distance->compare(target, query)); + } + + float normalize(float score) const + { + if (!Globals->scoreNormalization) return score; + return a * (score - b); } };