1 files changed, 2 insertions, 7 deletions
diff --git a/main.cc b/main.cc
index fc79c4b..eab058b 100644
--- a/main.cc
+++ b/main.cc
@@ -6,6 +6,7 @@
 #include "compute.h"
 #include "stitch.h"
 #include "explore.h"
+#include "utils.h"
 
 int main(int argc, char** argv)
 {
@@ -35,13 +36,7 @@ int main(int argc, char** argv)
   {
     Mat frame, frame_double;
     cap >> frame;
-    frame.convertTo(frame_double, CV_32FC3, 1.0/255.);
-    Mat aligned (frame_double.rows, frame_double.cols, CV_32FC4);
-    Mat channel4 = Mat::ones(frame_double.rows, frame_double.cols, CV_32F);
-    Mat in[] = {frame_double, channel4};
-    int from_to[] = { 0,0,  1,1,  2,2,  3,3 };
-    cv::mixChannels (in, 2, &aligned, 1, from_to, 4);
-    frames[i] = aligned.clone();
+    frames[i] = rgb_to_cspace4c(frame).clone();
   }
 
   vector<vector<Mat> > downscaled;