Function TEST

test/harris.cpp:170–222

using af::loadImage;

TEST(FloatHarris, CPP) {
    UNSUPPORTED_BACKEND(AF_BACKEND_ONEAPI);
    IMAGEIO_ENABLED_CHECK();

    vector<dim4> inDims;
    vector<string> inFiles;
    vector<vector<float>> gold;

    readImageTests(string(TEST_DIR "/harris/square_0_3.test"), inDims, inFiles,
                   gold);
    inFiles[0].insert(0, string(TEST_DIR "/harris/"));

    array in = loadImage(inFiles[0].c_str(), false);

    features out = harris(in, 500, 1e5f, 0.0f, 3, 0.04f);

    vector<float> outX(gold[0].size());
    vector<float> outY(gold[1].size());
    vector<float> outScore(gold[2].size());
    vector<float> outOrientation(gold[3].size());
    vector<float> outSize(gold[4].size());
    out.getX().host(&outX.front());
    out.getY().host(&outY.front());
    out.getScore().host(&outScore.front());
    out.getOrientation().host(&outOrientation.front());
    out.getSize().host(&outSize.front());

    vector<feat_t> out_feat;
    array_to_feat(out_feat, &outX.front(), &outY.front(), &outScore.front(),
                  &outOrientation.front(), &outSize.front(),
                  out.getNumFeatures());

    vector<feat_t> gold_feat;
    array_to_feat(gold_feat, &gold[0].front(), &gold[1].front(),
                  &gold[2].front(), &gold[3].front(), &gold[4].front(),
                  gold[0].size());

    std::sort(out_feat.begin(), out_feat.end(), feat_cmp);
    std::sort(gold_feat.begin(), gold_feat.end(), feat_cmp);

    for (unsigned elIter = 0; elIter < out.getNumFeatures(); elIter++) {
        ASSERT_EQ(out_feat[elIter].f[0], gold_feat[elIter].f[0])
            << "at: " << elIter << endl;
        ASSERT_EQ(out_feat[elIter].f[1], gold_feat[elIter].f[1])
            << "at: " << elIter << endl;
        ASSERT_LE(fabs(out_feat[elIter].f[2] - gold_feat[elIter].f[2]), 1e2)
            << "at: " << elIter << endl;
        ASSERT_EQ(out_feat[elIter].f[3], gold_feat[elIter].f[3])
            << "at: " << elIter << endl;
        ASSERT_EQ(out_feat[elIter].f[4], gold_feat[elIter].f[4])
            << "at: " << elIter << endl;
    }
}