Add a 'byRow' option to Normalize, to operate on rows of a matrix independently

Charles Otto
1 parent bbc08be4
Showing 1 changed file with 13 additions and 1 deletions
openbr/plugins/normalize.cpp
@@ -58,6 +58,9 @@ class NormalizeTransform : public UntrainableTransform
     Q_ENUMS(NormType)
     Q_PROPERTY(NormType normType READ get_normType WRITE set_normType RESET reset_normType STORED false)
+    Q_PROPERTY(bool ByRow READ get_ByRow WRITE set_ByRow RESET reset_ByRow STORED false)
+    BR_PROPERTY(bool, ByRow, false)
+
 public:
     /*!< */
     enum NormType { Inf = NORM_INF,
@@ -69,7 +72,16 @@ private:
     void project(const Template &src, Template &dst) const
     {
-        normalize(src, dst, 1, 0, normType, CV_32F);
+        if (!ByRow) normalize(src, dst, 1, 0, normType, CV_32F);
+        else {
+            dst = src;
+            for (int i=0; i<dst.m().rows; i++) {
+                Mat temp;
+                cv::normalize(dst.m().row(i), temp, 1, 0, normType);
+                temp.copyTo(dst.m().row(i));
+            }
+        }
+
     }
 };