OpenCV/opencv_contrib/modules/dnn_superres/perf/perf_dnn_superres.cpp

// 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 "perf_precomp.hpp"

using namespace std;
using namespace cv;
using namespace perf;

namespace opencv_test { namespace {

typedef perf::TestBaseWithParam<tuple<tuple<string,string,int>, string> > dnn_superres;

#define MODEL testing::Values(tuple<string,string,int> {"espcn","ESPCN_x2.pb",2}, \
                                tuple<string,string,int> {"lapsrn","LapSRN_x4.pb",4})
#define IMAGES testing::Values("cv/dnn_superres/butterfly.png", "cv/shared/baboon.png", "cv/shared/lena.png")

const string TEST_DIR = "cv/dnn_superres";

PERF_TEST_P(dnn_superres, upsample, testing::Combine(MODEL, IMAGES))
{
    tuple<string,string,int> model = get<0>( GetParam() );
    string image_name = get<1>( GetParam() );

    string model_name = get<0>(model);
    string model_filename = get<1>(model);
    int scale = get<2>(model);

    string model_path = cvtest::findDataFile(TEST_DIR + "/" + model_filename);
    string image_path = cvtest::findDataFile(image_name);

    DnnSuperResImpl sr;
    sr.readModel(model_path);
    sr.setModel(model_name, scale);

    Mat img = imread(image_path);
    ASSERT_FALSE(img.empty()) << image_path;

    Mat result;

    TEST_CYCLE() { sr.upsample(img, result); }

    ASSERT_FALSE(result.empty());

    SANITY_CHECK_NOTHING();
}

}}