109 lines
3.9 KiB
C++
109 lines
3.9 KiB
C++
|
// This file is part of OpenCV project.
|
||
|
|
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
||
|
|
// of this distribution and at http://opencv.org/license.html.
|
||
|
|
#include "test_precomp.hpp"
|
||
|
|
|
||
|
|
namespace opencv_test { namespace {
|
||
|
|
|
||
|
|
Mat testOilPainting(Mat imgSrc, int halfSize, int dynRatio, int colorSpace)
|
||
|
|
{
|
||
|
|
vector<int> histogramme;
|
||
|
|
vector<Vec3f> moyenneRGB;
|
||
|
|
Mat dst(imgSrc.size(), imgSrc.type());
|
||
|
|
Mat lum;
|
||
|
|
if (imgSrc.channels() != 1)
|
||
|
|
{
|
||
|
|
cvtColor(imgSrc, lum, colorSpace);
|
||
|
|
if (lum.channels() > 1)
|
||
|
|
{
|
||
|
|
extractChannel(lum, lum, 0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
lum = imgSrc.clone();
|
||
|
|
lum = lum / dynRatio;
|
||
|
|
if (dst.channels() == 3)
|
||
|
|
for (int y = 0; y < imgSrc.rows; y++)
|
||
|
|
{
|
||
|
|
Vec3b *vDst = dst.ptr<Vec3b>(y);
|
||
|
|
for (int x = 0; x < imgSrc.cols; x++, vDst++) //for each pixel
|
||
|
|
{
|
||
|
|
Mat mask(lum.size(), CV_8UC1, Scalar::all(0));
|
||
|
|
Rect r(Point(x - halfSize, y - halfSize), Size(2 * halfSize + 1, 2 * halfSize + 1));
|
||
|
|
r = r & Rect(Point(0, 0), lum.size());
|
||
|
|
mask(r).setTo(255);
|
||
|
|
int histSize[] = { 256 };
|
||
|
|
float hranges[] = { 0, 256 };
|
||
|
|
const float* ranges[] = { hranges };
|
||
|
|
Mat hist;
|
||
|
|
int channels[] = { 0 };
|
||
|
|
calcHist(&lum, 1, channels, mask, hist, 1, histSize, ranges, true, false);
|
||
|
|
double maxVal = 0;
|
||
|
|
Point pMin, pMax;
|
||
|
|
minMaxLoc(hist, 0, &maxVal, &pMin, &pMax);
|
||
|
|
mask.setTo(0, lum != static_cast<int>(pMax.y));
|
||
|
|
Scalar v = mean(imgSrc, mask);
|
||
|
|
*vDst = Vec3b(static_cast<uchar>(v[0]), static_cast<uchar>(v[1]), static_cast<uchar>(v[2]));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
for (int y = 0; y < imgSrc.rows; y++)
|
||
|
|
{
|
||
|
|
uchar *vDst = dst.ptr<uchar>(y);
|
||
|
|
for (int x = 0; x < imgSrc.cols; x++, vDst++) //for each pixel
|
||
|
|
{
|
||
|
|
Mat mask(lum.size(), CV_8UC1, Scalar::all(0));
|
||
|
|
Rect r(Point(x - halfSize, y - halfSize), Size(2 * halfSize + 1, 2 * halfSize + 1));
|
||
|
|
r = r & Rect(Point(0, 0), lum.size());
|
||
|
|
mask(r).setTo(255);
|
||
|
|
int histSize[] = { 256 };
|
||
|
|
float hranges[] = { 0, 256 };
|
||
|
|
const float* ranges[] = { hranges };
|
||
|
|
Mat hist;
|
||
|
|
int channels[] = { 0 };
|
||
|
|
calcHist(&lum, 1, channels, mask, hist, 1, histSize, ranges, true, false);
|
||
|
|
double maxVal = 0;
|
||
|
|
Point pMin, pMax;
|
||
|
|
minMaxLoc(hist, 0, &maxVal, &pMin, &pMax);
|
||
|
|
mask.setTo(0, lum != static_cast<int>(pMax.y));
|
||
|
|
Scalar v = mean(imgSrc, mask);
|
||
|
|
*vDst = static_cast<uchar>(v[0]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return dst;
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(xphoto_oil_painting, regression)
|
||
|
|
{
|
||
|
|
string folder = string(cvtest::TS::ptr()->get_data_path()) + "cv/inpaint/";
|
||
|
|
Mat orig = imread(folder+"exp1.png", IMREAD_COLOR);
|
||
|
|
ASSERT_TRUE(!orig.empty());
|
||
|
|
resize(orig, orig, Size(100, 100));
|
||
|
|
Mat dst1, dst2, dd;
|
||
|
|
xphoto::oilPainting(orig, dst1, 3, 5, COLOR_BGR2GRAY);
|
||
|
|
dst2 = testOilPainting(orig, 3, 5, COLOR_BGR2GRAY);
|
||
|
|
absdiff(dst1, dst2, dd);
|
||
|
|
vector<Mat> plane;
|
||
|
|
split(dd, plane);
|
||
|
|
for (auto p : plane)
|
||
|
|
{
|
||
|
|
double maxVal;
|
||
|
|
Point pIdx;
|
||
|
|
minMaxLoc(p, NULL, &maxVal, NULL, &pIdx);
|
||
|
|
ASSERT_LE(p.at<uchar>(pIdx), 2);
|
||
|
|
}
|
||
|
|
Mat orig2 = imread(folder + "exp1.png",IMREAD_GRAYSCALE);
|
||
|
|
ASSERT_TRUE(!orig2.empty());
|
||
|
|
resize(orig2, orig2, Size(100, 100));
|
||
|
|
Mat dst3, dst4, ddd;
|
||
|
|
xphoto::oilPainting(orig2, dst3, 3, 5, COLOR_BGR2GRAY);
|
||
|
|
dst4 = testOilPainting(orig2, 3, 5, COLOR_BGR2GRAY);
|
||
|
|
absdiff(dst3, dst4, ddd);
|
||
|
|
double maxVal;
|
||
|
|
Point pIdx;
|
||
|
|
minMaxLoc(ddd, NULL, &maxVal, NULL, &pIdx);
|
||
|
|
ASSERT_LE(ddd.at<uchar>(pIdx), 2);
|
||
|
|
}
|
||
|
|
|
||
|
|
}} // namespace
|