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.
61 lines
2.4 KiB
61 lines
2.4 KiB
// 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(); |
|
} |
|
} |
|
|
|
}}
|
|
|