opencv_contrib/modules/ximgproc/test/test_structured_edge_detection.cpp

#include "test_precomp.hpp"

namespace cvtest
{

TEST(ximpgroc_StructuredEdgeDetection, regression)
{
    cv::String dir = cvtest::TS::ptr()->get_data_path();
    int nTests = 12;
    float threshold = 0.01f;

    cv::String modelName = dir + "model.yml.gz";
    cv::Ptr<cv::ximgproc::StructuredEdgeDetection> pDollar =
        cv::ximgproc::createStructuredEdgeDetection(modelName);

    for (int i = 0; i < nTests; ++i)
    {
        cv::String srcName = dir + cv::format( "sources/%02d.png", i + 1);
        cv::Mat src = cv::imread( srcName, 1 );

        cv::String previousResultName = dir + cv::format( "results/%02d.png", i + 1 );
        cv::Mat previousResult = cv::imread( previousResultName, 0 );
        previousResult.convertTo( previousResult, cv::DataType<float>::type, 1/255.0 );

        src.convertTo( src, cv::DataType<float>::type, 1/255.0 );

        cv::Mat currentResult( src.size(), src.type() );
        pDollar->detectEdges( src, currentResult );

        cv::Mat sqrError = ( currentResult - previousResult )
            .mul( currentResult - previousResult );
        cv::Scalar mse = cv::sum(sqrError) / cv::Scalar::all( double( sqrError.total() ) );

        EXPECT_LE( mse[0], threshold );
    }
}

}