1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
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();
}
}
}}
|
