add to mcc detect and infer test

Author: AleksandrPanov

modules/mcc/test/test_mcc.cpp

@@ -81,5 +81,54 @@ TEST(CV_mccRunCCheckerDetectorBasic, accuracy_VINYL18)
     runCCheckerDetectorBasic("VINYL18.png", VINYL18);
 }
 
+TEST(CV_mcc_ccm_test, detectAndInfer)
+{
+    string path = cvtest::findDataFile("mcc/mcc_ccm_test.jpg");
+    Mat img = imread(path, IMREAD_COLOR);
+    Ptr<CCheckerDetector> detector = CCheckerDetector::create();
+
+    // detect MCC24 board
+    ASSERT_TRUE(detector->process(img, (TYPECHART)0, 1, true));
+
+    // read gold CCM
+    path = cvtest::findDataFile("mcc/mcc_ccm_test.yml");
+    FileStorage fs(path, FileStorage::READ);
+    ASSERT_TRUE(fs.isOpened());
+    FileNode node = fs["ccm"];
+    ASSERT_FALSE(node.empty());
+    Mat gold_ccm;
+    node >> gold_ccm;
+    fs.release();
+
+    // compute CCM
+    Ptr<CChecker> checker = detector->getBestColorChecker();
+    Mat chartsRGB = checker->getChartsRGB();
+    Mat src = chartsRGB.col(1).clone().reshape(3, chartsRGB.rows/3) / 255.;
+    ColorCorrectionModel model(src, COLORCHECKER_Macbeth);
+    model.run();
+    Mat ccm = model.getCCM();
+    // check CCM
+    ASSERT_MAT_NEAR(gold_ccm, ccm, 1e-12);
+
+    const double gold_loss = 4.6386569120323129;
+    const double loss = model.getLoss();
+    // check loss
+    EXPECT_NEAR(gold_loss, loss, 1e-12);
+
+    // read gold calibrate img
+    path = cvtest::findDataFile("mcc/mcc_ccm_test_res.png");
+    Mat gold_img = imread(path);
+
+    // compute calibrate image
+    Mat calibratedImage;
+    cvtColor(img, calibratedImage, COLOR_BGR2RGB);
+    calibratedImage.convertTo(calibratedImage, CV_64F, 1. / 255.);
+    calibratedImage = model.infer(calibratedImage)*255.;
+    calibratedImage.convertTo(calibratedImage, CV_8UC3);
+    cvtColor(calibratedImage, calibratedImage, COLOR_RGB2BGR);
+    // check calibrated image
+    EXPECT_MAT_NEAR(gold_img, calibratedImage, 1e-12);
+}
+
 } // namespace
 } // namespace opencv_test