/** * Lubricant Detecter * Copyright (C) 2021 Carl Klemm * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License * version 3 as published by the Free Software Foundation. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, * Boston, MA 02110-1301, USA. */ #pragma once #include #include #include inline bool normalize(cv::Mat& image, const cv::Mat& referance) { cv::Mat labReferance; cv::cvtColor(referance, labReferance, cv::COLOR_BGR2Lab); std::vector labPlanesRef(3); cv::split(labReferance, labPlanesRef); cv::Scalar mean = cv::mean(labPlanesRef[0]); labPlanesRef[0].convertTo(labPlanesRef[0], CV_16SC1); labPlanesRef[0] = labPlanesRef[0] - cv::Scalar(mean); cv::Mat labImage; cv::cvtColor(image, labImage, cv::COLOR_BGR2Lab); std::vector labPlanes(3); cv::split(labImage, labPlanes); cv::subtract(labPlanes[0], labPlanesRef[0], labPlanes[0], cv::noArray(), CV_8UC1); cv::merge(labPlanes, labImage); cv::cvtColor(labImage, image, cv::COLOR_Lab2BGR); return true; }