Home | History | Annotate | Download | only in test 
      1 /*M///////////////////////////////////////////////////////////////////////////////////////
      2 //
      3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
      4 //
      5 //  By downloading, copying, installing or using the software you agree to this license.
      6 //  If you do not agree to this license, do not download, install,
      7 //  copy or use the software.
      8 //
      9 //
     10 //                           License Agreement
     11 //                For Open Source Computer Vision Library
     12 //
     13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
     14 // Copyright (C) 2009, Willow Garage Inc., all rights reserved.
     15 // Third party copyrights are property of their respective owners.
     16 //
     17 // Redistribution and use in source and binary forms, with or without modification,
     18 // are permitted provided that the following conditions are met:
     19 //
     20 //   * Redistribution's of source code must retain the above copyright notice,
     21 //     this list of conditions and the following disclaimer.
     22 //
     23 //   * Redistribution's in binary form must reproduce the above copyright notice,
     24 //     this list of conditions and the following disclaimer in the documentation
     25 //     and/or other materials provided with the distribution.
     26 //
     27 //   * The name of the copyright holders may not be used to endorse or promote products
     28 //     derived from this software without specific prior written permission.
     29 //
     30 // This software is provided by the copyright holders and contributors "as is" and
     31 // any express or implied warranties, including, but not limited to, the implied
     32 // warranties of merchantability and fitness for a particular purpose are disclaimed.
     33 // In no event shall the Intel Corporation or contributors be liable for any direct,
     34 // indirect, incidental, special, exemplary, or consequential damages
     35 // (including, but not limited to, procurement of substitute goods or services;
     36 // loss of use, data, or profits; or business interruption) however caused
     37 // and on any theory of liability, whether in contract, strict liability,
     38 // or tort (including negligence or otherwise) arising in any way out of
     39 // the use of this software, even if advised of the possibility of such damage.
     40 //
     41 //M*/
     42 
     43 #include "test_precomp.hpp"
     44 #include <string>
     45 
     46 using namespace std;
     47 using namespace cv;
     48 
     49 class CV_InpaintTest : public cvtest::BaseTest
     50 {
     51 public:
     52     CV_InpaintTest();
     53     ~CV_InpaintTest();
     54 protected:
     55     void run(int);
     56 };
     57 
     58 CV_InpaintTest::CV_InpaintTest()
     59 {
     60 }
     61 CV_InpaintTest::~CV_InpaintTest() {}
     62 
     63 void CV_InpaintTest::run( int )
     64 {
     65     string folder = string(ts->get_data_path()) + "inpaint/";
     66     Mat orig = imread(folder + "orig.png");
     67     Mat exp1 = imread(folder + "exp1.png");
     68     Mat exp2 = imread(folder + "exp2.png");
     69     Mat mask = imread(folder + "mask.png");
     70 
     71     if (orig.empty() || exp1.empty() || exp2.empty() || mask.empty())
     72     {
     73         ts->set_failed_test_info( cvtest::TS::FAIL_INVALID_TEST_DATA );
     74         return;
     75     }
     76 
     77     Mat inv_mask;
     78     mask.convertTo(inv_mask, CV_8UC3, -1.0, 255.0);
     79 
     80     Mat mask1ch;
     81     cv::cvtColor(mask, mask1ch, COLOR_BGR2GRAY);
     82 
     83     Mat test = orig.clone();
     84     test.setTo(Scalar::all(255), mask1ch);
     85 
     86     Mat res1, res2;
     87     inpaint( test, mask1ch, res1, 5, INPAINT_NS );
     88     inpaint( test, mask1ch, res2, 5, INPAINT_TELEA );
     89 
     90     Mat diff1, diff2;
     91     absdiff( orig, res1, diff1 );
     92     absdiff( orig, res2, diff2 );
     93 
     94     double n1 = cvtest::norm(diff1.reshape(1), NORM_INF, inv_mask.reshape(1));
     95     double n2 = cvtest::norm(diff2.reshape(1), NORM_INF, inv_mask.reshape(1));
     96 
     97     if (n1 != 0 || n2 != 0)
     98     {
     99         ts->set_failed_test_info( cvtest::TS::FAIL_MISMATCH );
    100         return;
    101     }
    102 
    103     absdiff( exp1, res1, diff1 );
    104     absdiff( exp2, res2, diff2 );
    105 
    106     n1 = cvtest::norm(diff1.reshape(1), NORM_INF, mask.reshape(1));
    107     n2 = cvtest::norm(diff2.reshape(1), NORM_INF, mask.reshape(1));
    108 
    109     const int jpeg_thres = 3;
    110     if (n1 > jpeg_thres || n2 > jpeg_thres)
    111     {
    112         ts->set_failed_test_info( cvtest::TS::FAIL_BAD_ACCURACY );
    113         return;
    114     }
    115 
    116     ts->set_failed_test_info(cvtest::TS::OK);
    117 }
    118 
    119 TEST(Photo_Inpaint, regression) { CV_InpaintTest test; test.safe_run(); }
    120