vendor: OpenCV 5.0.0 snapshot at 755e50675d97db9b7d449d8bd6b09888646f6c6e
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
// This file is part of OpenCV project.
|
||||
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
||||
// of this distribution and at http://opencv.org/license.html.
|
||||
|
||||
#include "perf_precomp.hpp"
|
||||
|
||||
namespace opencv_test { namespace {
|
||||
static void drawCircle(cv::Mat& dst, const cv::Vec3i& circle, bool fill)
|
||||
{
|
||||
dst.setTo(Scalar::all(0));
|
||||
cv::circle(dst, Point2i(circle[0], circle[1]), circle[2], Scalar::all(255), fill ? -1 : 1, cv::LINE_AA);
|
||||
}
|
||||
|
||||
DEF_PARAM_TEST(Sz_Depth, Size, MatDepth);
|
||||
PERF_TEST_P(Sz_Depth, SpatialMoments, Combine(CUDA_TYPICAL_MAT_SIZES, Values(MatDepth(CV_32F), MatDepth((CV_64F)))))
|
||||
{
|
||||
const cv::Size size = GET_PARAM(0);
|
||||
const int momentsType = GET_PARAM(1);
|
||||
Mat imgHost(size, CV_8U);
|
||||
const Vec3i circle(size.width / 2, size.height / 2, static_cast<int>(static_cast<float>(size.width / 2) * 0.9));
|
||||
drawCircle(imgHost, circle, true);
|
||||
if (PERF_RUN_CUDA()) {
|
||||
const MomentsOrder order = MomentsOrder::THIRD_ORDER_MOMENTS;
|
||||
const int nMoments = numMoments(order);
|
||||
GpuMat momentsDevice(1, nMoments, momentsType);
|
||||
const GpuMat imgDevice(imgHost);
|
||||
TEST_CYCLE() cuda::spatialMoments(imgDevice, momentsDevice, false, order, momentsType);
|
||||
SANITY_CHECK_NOTHING();
|
||||
}
|
||||
else {
|
||||
cv::Moments momentsHost;
|
||||
TEST_CYCLE() momentsHost = cv::moments(imgHost, false);
|
||||
SANITY_CHECK_NOTHING();
|
||||
}
|
||||
}
|
||||
|
||||
PERF_TEST_P(Sz_Depth, Moments, Combine(CUDA_TYPICAL_MAT_SIZES, Values(MatDepth(CV_32F), MatDepth(CV_64F))))
|
||||
{
|
||||
const cv::Size size = GET_PARAM(0);
|
||||
const int momentsType = GET_PARAM(1);
|
||||
Mat imgHost(size, CV_8U);
|
||||
const Vec3i circle(size.width / 2, size.height / 2, static_cast<int>(static_cast<float>(size.width / 2) * 0.9));
|
||||
drawCircle(imgHost, circle, true);
|
||||
if (PERF_RUN_CUDA()) {
|
||||
const MomentsOrder order = MomentsOrder::THIRD_ORDER_MOMENTS;
|
||||
const int nMoments = numMoments(order);
|
||||
setBufferPoolUsage(true);
|
||||
setBufferPoolConfig(getDevice(), nMoments * ((momentsType == CV_64F) ? sizeof(double) : sizeof(float)), 1);
|
||||
const GpuMat imgDevice(imgHost);
|
||||
cv::Moments momentsHost;
|
||||
TEST_CYCLE() momentsHost = cuda::moments(imgDevice, false, order, momentsType);
|
||||
SANITY_CHECK_NOTHING();
|
||||
}
|
||||
else {
|
||||
cv::Moments momentsHost;
|
||||
TEST_CYCLE() momentsHost = cv::moments(imgHost, false);
|
||||
SANITY_CHECK_NOTHING();
|
||||
}
|
||||
}
|
||||
|
||||
}}
|
||||
Reference in New Issue
Block a user