You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.9 KiB
60 lines
1.9 KiB
/* |
|
* 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<cv::Size, MatType, int /*iterations*/, float /*epsilon*/, Size /*winSize*/> MeanShiftPerfParams; |
|
typedef perf::TestBaseWithParam<MeanShiftPerfParams> MeanShiftPerfTest; |
|
|
|
PERF_TEST_P(MeanShiftPerfTest, run, |
|
::testing::Combine(::testing::Values(Size(128, 128), Size(640, 480), Size(800, 600)), |
|
::testing::Values(CV_8U, CV_32S, CV_32F), // type |
|
::testing::Values(2, 10, 100), // nIterations |
|
::testing::Values(0.01f, 0.1f, 1.f, 10.f), // epsilon |
|
::testing::Values(Size(8, 8), Size(13, 48), Size(64, 64)) // window size |
|
) |
|
) |
|
{ |
|
auto p = GetParam(); |
|
cv::Size size = std::get<0>(p); |
|
MatType type = std::get<1>(p); |
|
int iters = std::get<2>(p); |
|
float eps = std::get<3>(p); |
|
Size winSize = std::get<4>(p); |
|
|
|
RNG& rng = cv::theRNG(); |
|
|
|
const int nPts = 20; |
|
Mat ptsMap(size, CV_8UC1, Scalar(255)); |
|
for(size_t i = 0; i < nPts; ++i) |
|
{ |
|
ptsMap.at<uchar>(rng() % size.height, rng() % size.width) = 0; |
|
} |
|
Mat distTrans(size, CV_8UC1); |
|
cv::distanceTransform(ptsMap, distTrans, DIST_L2, DIST_MASK_PRECISE); |
|
Mat vsrc = 255 - distTrans; |
|
Mat src; |
|
vsrc.convertTo(src, type); |
|
|
|
Point startPt(rng() % (size.width - winSize.width), |
|
rng() % (size.height - winSize.height)); |
|
Rect startRect(startPt, winSize); |
|
|
|
cv::TermCriteria termCrit( TermCriteria::EPS + TermCriteria::MAX_ITER, iters, eps); |
|
|
|
Rect window = startRect; |
|
while(next()) |
|
{ |
|
startTimer(); |
|
cv::fastcv::meanShift(src, window, termCrit); |
|
stopTimer(); |
|
} |
|
|
|
SANITY_CHECK_NOTHING(); |
|
} |
|
|
|
} // namespace
|
|
|