From b1cd12990e8408e1d5095bf21e98437f1e64c0e8 Mon Sep 17 00:00:00 2001
From: Nicolas Pope <nwpope@utu.fi>
Date: Tue, 18 Jun 2019 10:13:13 +0300
Subject: [PATCH] Add local whitebalance
---
components/rgbd-sources/src/local.cpp | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/components/rgbd-sources/src/local.cpp b/components/rgbd-sources/src/local.cpp
index 63257bad5..a0d6334b3 100644
--- a/components/rgbd-sources/src/local.cpp
+++ b/components/rgbd-sources/src/local.cpp
@@ -11,6 +11,7 @@
#include "local.hpp"
#include <opencv2/core.hpp>
#include <opencv2/opencv.hpp>
+#include <opencv2/xphoto.hpp>
using ftl::rgbd::detail::LocalSource;
using cv::Mat;
@@ -269,6 +270,11 @@ bool LocalSource::get(cv::Mat &l, cv::Mat &r) {
0, 0, cv::INTER_LINEAR);
}
+ cv::Ptr<cv::xphoto::WhiteBalancer> wb;
+ wb = cv::xphoto::createSimpleWB();
+ wb->balanceWhite(l, l);
+ wb->balanceWhite(r, r);
+
if (flip_v_) {
Mat tl, tr;
cv::flip(l, tl, 0);
--
GitLab