diff --git a/components/rgbd-sources/src/local.cpp b/components/rgbd-sources/src/local.cpp index 63257bad5844812eda20835ea4ebe022b4e58a2f..a0d6334b357e8c51243fa2922123333092640e23 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);