Gitea Mirror Bot
77 lines
2.4 KiB
C++
77 lines
2.4 KiB
C++
/*
|
|
* Copyright (c) 2024 Qualcomm Innovation Center, Inc. All rights reserved.
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
#include "perf_precomp.hpp"
|
|
|
|
namespace opencv_test {
|
|
|
|
typedef std::tuple<bool /*useFloat*/, int /*nLevels*/, bool /*scaleBy2*/> PyramidTestParams;
|
|
class PyramidTest : public ::perf::TestBaseWithParam<PyramidTestParams> { };
|
|
|
|
PERF_TEST_P(PyramidTest, checkAllVersions, // version, useFloat, nLevels
|
|
::testing::Values(
|
|
PyramidTestParams { true, 2, true}, PyramidTestParams { true, 3, true}, PyramidTestParams { true, 4, true},
|
|
PyramidTestParams {false, 2, true}, PyramidTestParams {false, 3, true}, PyramidTestParams {false, 4, true},
|
|
PyramidTestParams {false, 2, false}, PyramidTestParams {false, 3, false}, PyramidTestParams {false, 4, false}
|
|
))
|
|
{
|
|
auto par = GetParam();
|
|
|
|
bool useFloat = std::get<0>(par);
|
|
int nLevels = std::get<1>(par);
|
|
bool scaleBy2 = std::get<2>(par);
|
|
|
|
cv::Mat src = imread(cvtest::findDataFile("cv/shared/baboon.png"), cv::IMREAD_GRAYSCALE);
|
|
|
|
if (useFloat)
|
|
{
|
|
cv::Mat f;
|
|
src.convertTo(f, CV_32F);
|
|
src = f;
|
|
}
|
|
|
|
while(next())
|
|
{
|
|
std::vector<cv::Mat> pyr;
|
|
startTimer();
|
|
cv::fastcv::buildPyramid(src, pyr, nLevels, scaleBy2);
|
|
stopTimer();
|
|
}
|
|
|
|
SANITY_CHECK_NOTHING();
|
|
}
|
|
|
|
|
|
typedef std::tuple<MatType, size_t> SobelPyramidTestParams;
|
|
class SobelPyramidTest : public ::perf::TestBaseWithParam<SobelPyramidTestParams> {};
|
|
|
|
PERF_TEST_P(SobelPyramidTest, checkAllTypes,
|
|
::testing::Combine(::testing::Values(CV_8S, CV_16S, CV_32F),
|
|
::testing::Values(3, 6)))
|
|
{
|
|
auto p = GetParam();
|
|
int type = std::get<0>(p);
|
|
size_t nLevels = std::get<1>(p);
|
|
|
|
// NOTE: test files should be manually loaded to folder on a device, for example like this:
|
|
// adb push fastcv/misc/bilateral_recursive/ /sdcard/testdata/fastcv/bilateral/
|
|
cv::Mat src = imread(cvtest::findDataFile("cv/shared/baboon.png"), cv::IMREAD_GRAYSCALE);
|
|
|
|
std::vector<cv::Mat> pyr;
|
|
cv::fastcv::buildPyramid(src, pyr, nLevels);
|
|
|
|
while(next())
|
|
{
|
|
std::vector<cv::Mat> pyrDx, pyrDy;
|
|
startTimer();
|
|
cv::fastcv::sobelPyramid(pyr, pyrDx, pyrDy, type);
|
|
stopTimer();
|
|
}
|
|
|
|
SANITY_CHECK_NOTHING();
|
|
}
|
|
|
|
} // namespace
|