Do not flatten to a single Mat in CatCols

Austin Van Blanton
1 parent 068b72ad
Showing 1 changed file with 8 additions and 6 deletions
openbr/plugins/regions.cpp
@@ -141,13 +141,15 @@ class CatColsTransform : public UntrainableMetaTransform
     void project(const Template &src, Template &dst) const
     {
+        int half = src.size()/2;
+        for (int i=0; i<half; i++) {
+            Mat first = src[i];
+            Mat second = src[half+i];
+            Mat both;
+            hconcat(first, second, both);
+            dst.append(both);
+        }
         dst.file = src.file;
-        Mat m = OpenCVUtils::toMatByRow(src);
-        // right now this just splits src in half and joins them horizontally
-        // TODO: add partitions parameter for more than a single split
-        Mat first = m.rowRange(Range(0, m.rows/2));
-        Mat second = m.rowRange(Range(m.rows/2, m.rows));
-        hconcat(first, second, dst);
     }
 };