Function cannyImageOtsuTest

test/canny.cpp:96–159

template<typename T>
void cannyImageOtsuTest(string pTestFile, bool isColor) {
    SUPPORTED_TYPE_CHECK(T);
    IMAGEIO_ENABLED_CHECK();

    using af::dim4;

    vector<dim4> inDims;
    vector<string> inFiles;
    vector<dim_t> outSizes;
    vector<string> outFiles;

    readImageTests(pTestFile, inDims, inFiles, outSizes, outFiles);

    size_t testCount = inDims.size();

    for (size_t testId = 0; testId < testCount; ++testId) {
        af_array _inArray  = 0;
        af_array inArray   = 0;
        af_array _outArray = 0;
        af_array cstArray  = 0;
        af_array mulArray  = 0;
        af_array outArray  = 0;
        af_array goldArray = 0;

        inFiles[testId].insert(0, string(TEST_DIR "/CannyEdgeDetector/"));
        outFiles[testId].insert(0, string(TEST_DIR "/CannyEdgeDetector/"));

        af_dtype type = (af_dtype)dtype_traits<T>::af_type;

        ASSERT_SUCCESS(
            af_load_image(&_inArray, inFiles[testId].c_str(), isColor));

        ASSERT_SUCCESS(af_cast(&inArray, _inArray, type));

        ASSERT_SUCCESS(
            af_load_image_native(&goldArray, outFiles[testId].c_str()));

        ASSERT_SUCCESS(af_canny(&_outArray, inArray,
                                            AF_CANNY_THRESHOLD_AUTO_OTSU,
                                            0.08, 0.32, 3, false));

        unsigned ndims = 0;
        dim_t dims[4];

        ASSERT_SUCCESS(af_get_numdims(&ndims, _outArray));
        ASSERT_SUCCESS(
            af_get_dims(dims, dims + 1, dims + 2, dims + 3, _outArray));

        ASSERT_SUCCESS(af_constant(&cstArray, 255.0, ndims, dims, f32));

        ASSERT_SUCCESS(af_mul(&mulArray, cstArray, _outArray, false));
        ASSERT_SUCCESS(af_cast(&outArray, mulArray, u8));

        ASSERT_IMAGES_NEAR(goldArray, outArray, 1.0e-3);

        ASSERT_SUCCESS(af_release_array(_inArray));
        ASSERT_SUCCESS(af_release_array(inArray));
        ASSERT_SUCCESS(af_release_array(cstArray));