2012-10-17 15:12:04 +08:00
|
|
|
/*M///////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
//
|
|
|
|
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
|
|
|
//
|
|
|
|
// By downloading, copying, installing or using the software you agree to this license.
|
|
|
|
// If you do not agree to this license, do not download, install,
|
|
|
|
// copy or use the software.
|
|
|
|
//
|
|
|
|
//
|
|
|
|
// License Agreement
|
|
|
|
// For Open Source Computer Vision Library
|
|
|
|
//
|
|
|
|
// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
|
|
|
|
// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
|
|
|
|
// Third party copyrights are property of their respective owners.
|
|
|
|
//
|
|
|
|
// Redistribution and use in source and binary forms, with or without modification,
|
|
|
|
// are permitted provided that the following conditions are met:
|
|
|
|
//
|
|
|
|
// * Redistribution's of source code must retain the above copyright notice,
|
|
|
|
// this list of conditions and the following disclaimer.
|
|
|
|
//
|
|
|
|
// * Redistribution's in binary form must reproduce the above copyright notice,
|
|
|
|
// this list of conditions and the following disclaimer in the documentation
|
|
|
|
// and/or other materials provided with the distribution.
|
|
|
|
//
|
|
|
|
// * The name of the copyright holders may not be used to endorse or promote products
|
|
|
|
// derived from this software without specific prior written permission.
|
|
|
|
//
|
|
|
|
// This software is provided by the copyright holders and contributors "as is" and
|
|
|
|
// any express or implied warranties, including, but not limited to, the implied
|
|
|
|
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
|
|
|
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
|
|
|
// indirect, incidental, special, exemplary, or consequential damages
|
|
|
|
// (including, but not limited to, procurement of substitute goods or services;
|
|
|
|
// loss of use, data, or profits; or business interruption) however caused
|
|
|
|
// and on any theory of liability, whether in contract, strict liability,
|
|
|
|
// or tort (including negligence or otherwise) arising in any way out of
|
|
|
|
// the use of this software, even if advised of the possibility of such damage.
|
|
|
|
//
|
|
|
|
//M*/
|
|
|
|
|
|
|
|
#include "precomp.hpp"
|
|
|
|
|
|
|
|
using namespace cv;
|
|
|
|
using namespace cv::gpu;
|
|
|
|
|
2013-06-04 17:32:35 +08:00
|
|
|
#if !defined (HAVE_CUDA) || !defined (HAVE_OPENCV_GPUARITHM) || defined (CUDA_DISABLER)
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
Ptr<gpu::TemplateMatching> cv::gpu::createTemplateMatching(int, int, Size) { throw_no_cuda(); return Ptr<gpu::TemplateMatching>(); }
|
2012-10-17 15:12:04 +08:00
|
|
|
|
|
|
|
#else
|
|
|
|
|
2013-04-04 19:36:22 +08:00
|
|
|
namespace cv { namespace gpu { namespace cudev
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
|
|
|
namespace match_template
|
|
|
|
{
|
|
|
|
void matchTemplateNaive_CCORR_8U(const PtrStepSzb image, const PtrStepSzb templ, PtrStepSzf result, int cn, cudaStream_t stream);
|
|
|
|
void matchTemplateNaive_CCORR_32F(const PtrStepSzb image, const PtrStepSzb templ, PtrStepSzf result, int cn, cudaStream_t stream);
|
|
|
|
|
|
|
|
void matchTemplateNaive_SQDIFF_8U(const PtrStepSzb image, const PtrStepSzb templ, PtrStepSzf result, int cn, cudaStream_t stream);
|
|
|
|
void matchTemplateNaive_SQDIFF_32F(const PtrStepSzb image, const PtrStepSzb templ, PtrStepSzf result, int cn, cudaStream_t stream);
|
|
|
|
|
|
|
|
void matchTemplatePrepared_SQDIFF_8U(int w, int h, const PtrStepSz<unsigned long long> image_sqsum, unsigned long long templ_sqsum, PtrStepSzf result,
|
|
|
|
int cn, cudaStream_t stream);
|
|
|
|
|
|
|
|
void matchTemplatePrepared_SQDIFF_NORMED_8U(int w, int h, const PtrStepSz<unsigned long long> image_sqsum, unsigned long long templ_sqsum, PtrStepSzf result,
|
|
|
|
int cn, cudaStream_t stream);
|
|
|
|
|
|
|
|
void matchTemplatePrepared_CCOFF_8U(int w, int h, const PtrStepSz<unsigned int> image_sum, unsigned int templ_sum, PtrStepSzf result, cudaStream_t stream);
|
|
|
|
void matchTemplatePrepared_CCOFF_8UC2(
|
|
|
|
int w, int h,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_r,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_g,
|
|
|
|
unsigned int templ_sum_r,
|
|
|
|
unsigned int templ_sum_g,
|
|
|
|
PtrStepSzf result, cudaStream_t stream);
|
|
|
|
void matchTemplatePrepared_CCOFF_8UC3(
|
|
|
|
int w, int h,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_r,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_g,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_b,
|
|
|
|
unsigned int templ_sum_r,
|
|
|
|
unsigned int templ_sum_g,
|
|
|
|
unsigned int templ_sum_b,
|
|
|
|
PtrStepSzf result, cudaStream_t stream);
|
|
|
|
void matchTemplatePrepared_CCOFF_8UC4(
|
|
|
|
int w, int h,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_r,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_g,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_b,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_a,
|
|
|
|
unsigned int templ_sum_r,
|
|
|
|
unsigned int templ_sum_g,
|
|
|
|
unsigned int templ_sum_b,
|
|
|
|
unsigned int templ_sum_a,
|
|
|
|
PtrStepSzf result, cudaStream_t stream);
|
|
|
|
|
|
|
|
|
|
|
|
void matchTemplatePrepared_CCOFF_NORMED_8U(
|
|
|
|
int w, int h, const PtrStepSz<unsigned int> image_sum,
|
|
|
|
const PtrStepSz<unsigned long long> image_sqsum,
|
|
|
|
unsigned int templ_sum, unsigned long long templ_sqsum,
|
|
|
|
PtrStepSzf result, cudaStream_t stream);
|
|
|
|
void matchTemplatePrepared_CCOFF_NORMED_8UC2(
|
|
|
|
int w, int h,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_r, const PtrStepSz<unsigned long long> image_sqsum_r,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_g, const PtrStepSz<unsigned long long> image_sqsum_g,
|
|
|
|
unsigned int templ_sum_r, unsigned long long templ_sqsum_r,
|
|
|
|
unsigned int templ_sum_g, unsigned long long templ_sqsum_g,
|
|
|
|
PtrStepSzf result, cudaStream_t stream);
|
|
|
|
void matchTemplatePrepared_CCOFF_NORMED_8UC3(
|
|
|
|
int w, int h,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_r, const PtrStepSz<unsigned long long> image_sqsum_r,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_g, const PtrStepSz<unsigned long long> image_sqsum_g,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_b, const PtrStepSz<unsigned long long> image_sqsum_b,
|
|
|
|
unsigned int templ_sum_r, unsigned long long templ_sqsum_r,
|
|
|
|
unsigned int templ_sum_g, unsigned long long templ_sqsum_g,
|
|
|
|
unsigned int templ_sum_b, unsigned long long templ_sqsum_b,
|
|
|
|
PtrStepSzf result, cudaStream_t stream);
|
|
|
|
void matchTemplatePrepared_CCOFF_NORMED_8UC4(
|
|
|
|
int w, int h,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_r, const PtrStepSz<unsigned long long> image_sqsum_r,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_g, const PtrStepSz<unsigned long long> image_sqsum_g,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_b, const PtrStepSz<unsigned long long> image_sqsum_b,
|
|
|
|
const PtrStepSz<unsigned int> image_sum_a, const PtrStepSz<unsigned long long> image_sqsum_a,
|
|
|
|
unsigned int templ_sum_r, unsigned long long templ_sqsum_r,
|
|
|
|
unsigned int templ_sum_g, unsigned long long templ_sqsum_g,
|
|
|
|
unsigned int templ_sum_b, unsigned long long templ_sqsum_b,
|
|
|
|
unsigned int templ_sum_a, unsigned long long templ_sqsum_a,
|
|
|
|
PtrStepSzf result, cudaStream_t stream);
|
|
|
|
|
|
|
|
void normalize_8U(int w, int h, const PtrStepSz<unsigned long long> image_sqsum,
|
|
|
|
unsigned long long templ_sqsum, PtrStepSzf result, int cn, cudaStream_t stream);
|
|
|
|
|
|
|
|
void extractFirstChannel_32F(const PtrStepSzb image, PtrStepSzf result, int cn, cudaStream_t stream);
|
|
|
|
}
|
|
|
|
}}}
|
|
|
|
|
|
|
|
namespace
|
|
|
|
{
|
|
|
|
// Evaluates optimal template's area threshold. If
|
|
|
|
// template's area is less than the threshold, we use naive match
|
|
|
|
// template version, otherwise FFT-based (if available)
|
|
|
|
int getTemplateThreshold(int method, int depth)
|
|
|
|
{
|
|
|
|
switch (method)
|
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
case TM_CCORR:
|
2012-10-17 15:12:04 +08:00
|
|
|
if (depth == CV_32F) return 250;
|
|
|
|
if (depth == CV_8U) return 300;
|
|
|
|
break;
|
2013-04-30 21:27:06 +08:00
|
|
|
|
|
|
|
case TM_SQDIFF:
|
2012-10-17 15:12:04 +08:00
|
|
|
if (depth == CV_8U) return 300;
|
|
|
|
break;
|
|
|
|
}
|
2013-04-30 21:27:06 +08:00
|
|
|
|
|
|
|
CV_Error(Error::StsBadArg, "unsupported match template mode");
|
2012-10-17 15:12:04 +08:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
///////////////////////////////////////////////////////////////
|
|
|
|
// CCORR_32F
|
|
|
|
|
|
|
|
class Match_CCORR_32F : public TemplateMatching
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
explicit Match_CCORR_32F(Size user_block_size);
|
|
|
|
|
|
|
|
void match(InputArray image, InputArray templ, OutputArray result, Stream& stream = Stream::Null());
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
private:
|
|
|
|
Ptr<gpu::Convolution> conv_;
|
|
|
|
GpuMat result_;
|
|
|
|
};
|
|
|
|
|
|
|
|
Match_CCORR_32F::Match_CCORR_32F(Size user_block_size)
|
|
|
|
{
|
|
|
|
conv_ = gpu::createConvolution(user_block_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Match_CCORR_32F::match(InputArray _image, InputArray _templ, OutputArray _result, Stream& _stream)
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
using namespace cv::gpu::cudev::match_template;
|
|
|
|
|
|
|
|
GpuMat image = _image.getGpuMat();
|
|
|
|
GpuMat templ = _templ.getGpuMat();
|
|
|
|
|
|
|
|
CV_Assert( image.depth() == CV_32F );
|
|
|
|
CV_Assert( image.type() == templ.type() );
|
|
|
|
CV_Assert( image.cols >= templ.cols && image.rows >= templ.rows );
|
|
|
|
|
|
|
|
cudaStream_t stream = StreamAccessor::getStream(_stream);
|
|
|
|
|
|
|
|
_result.create(image.rows - templ.rows + 1, image.cols - templ.cols + 1, CV_32FC1);
|
|
|
|
GpuMat result = _result.getGpuMat();
|
|
|
|
|
|
|
|
if (templ.size().area() < getTemplateThreshold(TM_CCORR, CV_32F))
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
matchTemplateNaive_CCORR_32F(image, templ, result, image.channels(), stream);
|
2012-10-17 15:12:04 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (image.channels() == 1)
|
2013-04-26 16:40:03 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
conv_->convolve(image.reshape(1), templ.reshape(1), result, true, _stream);
|
2013-04-26 16:40:03 +08:00
|
|
|
}
|
2012-10-17 15:12:04 +08:00
|
|
|
else
|
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
conv_->convolve(image.reshape(1), templ.reshape(1), result_, true, _stream);
|
|
|
|
extractFirstChannel_32F(result_, result, image.channels(), stream);
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
///////////////////////////////////////////////////////////////
|
|
|
|
// CCORR_8U
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
class Match_CCORR_8U : public TemplateMatching
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
public:
|
|
|
|
explicit Match_CCORR_8U(Size user_block_size) : match32F_(user_block_size)
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void match(InputArray image, InputArray templ, OutputArray result, Stream& stream = Stream::Null());
|
|
|
|
|
|
|
|
private:
|
|
|
|
GpuMat imagef_, templf_;
|
|
|
|
Match_CCORR_32F match32F_;
|
|
|
|
};
|
|
|
|
|
|
|
|
void Match_CCORR_8U::match(InputArray _image, InputArray _templ, OutputArray _result, Stream& stream)
|
|
|
|
{
|
|
|
|
using namespace cv::gpu::cudev::match_template;
|
|
|
|
|
|
|
|
GpuMat image = _image.getGpuMat();
|
|
|
|
GpuMat templ = _templ.getGpuMat();
|
|
|
|
|
|
|
|
CV_Assert( image.depth() == CV_8U );
|
|
|
|
CV_Assert( image.type() == templ.type() );
|
|
|
|
CV_Assert( image.cols >= templ.cols && image.rows >= templ.rows );
|
|
|
|
|
|
|
|
if (templ.size().area() < getTemplateThreshold(TM_CCORR, CV_8U))
|
|
|
|
{
|
|
|
|
_result.create(image.rows - templ.rows + 1, image.cols - templ.cols + 1, CV_32FC1);
|
|
|
|
GpuMat result = _result.getGpuMat();
|
|
|
|
|
2012-10-17 15:12:04 +08:00
|
|
|
matchTemplateNaive_CCORR_8U(image, templ, result, image.channels(), StreamAccessor::getStream(stream));
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
image.convertTo(imagef_, CV_32F, stream);
|
|
|
|
templ.convertTo(templf_, CV_32F, stream);
|
2013-04-16 21:43:49 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
match32F_.match(imagef_, templf_, _result, stream);
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
///////////////////////////////////////////////////////////////
|
|
|
|
// CCORR_NORMED_8U
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
class Match_CCORR_NORMED_8U : public TemplateMatching
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
public:
|
|
|
|
explicit Match_CCORR_NORMED_8U(Size user_block_size) : match_CCORR_(user_block_size)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
void match(InputArray image, InputArray templ, OutputArray result, Stream& stream = Stream::Null());
|
|
|
|
|
|
|
|
private:
|
|
|
|
Match_CCORR_8U match_CCORR_;
|
|
|
|
GpuMat image_sqsums_;
|
|
|
|
GpuMat intBuffer_;
|
|
|
|
};
|
|
|
|
|
|
|
|
void Match_CCORR_NORMED_8U::match(InputArray _image, InputArray _templ, OutputArray _result, Stream& stream)
|
|
|
|
{
|
|
|
|
using namespace cv::gpu::cudev::match_template;
|
|
|
|
|
|
|
|
GpuMat image = _image.getGpuMat();
|
|
|
|
GpuMat templ = _templ.getGpuMat();
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
CV_Assert( image.depth() == CV_8U );
|
|
|
|
CV_Assert( image.type() == templ.type() );
|
|
|
|
CV_Assert( image.cols >= templ.cols && image.rows >= templ.rows );
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
match_CCORR_.match(image, templ, _result, stream);
|
|
|
|
GpuMat result = _result.getGpuMat();
|
|
|
|
|
|
|
|
gpu::sqrIntegral(image.reshape(1), image_sqsums_, intBuffer_, stream);
|
|
|
|
|
|
|
|
unsigned long long templ_sqsum = (unsigned long long) gpu::sqrSum(templ.reshape(1))[0];
|
|
|
|
|
|
|
|
normalize_8U(templ.cols, templ.rows, image_sqsums_, templ_sqsum, result, image.channels(), StreamAccessor::getStream(stream));
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
///////////////////////////////////////////////////////////////
|
|
|
|
// SQDIFF_32F
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
class Match_SQDIFF_32F : public TemplateMatching
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
public:
|
|
|
|
void match(InputArray image, InputArray templ, OutputArray result, Stream& stream = Stream::Null());
|
|
|
|
};
|
|
|
|
|
|
|
|
void Match_SQDIFF_32F::match(InputArray _image, InputArray _templ, OutputArray _result, Stream& stream)
|
|
|
|
{
|
|
|
|
using namespace cv::gpu::cudev::match_template;
|
|
|
|
|
|
|
|
GpuMat image = _image.getGpuMat();
|
|
|
|
GpuMat templ = _templ.getGpuMat();
|
|
|
|
|
|
|
|
CV_Assert( image.depth() == CV_32F );
|
|
|
|
CV_Assert( image.type() == templ.type() );
|
|
|
|
CV_Assert( image.cols >= templ.cols && image.rows >= templ.rows );
|
|
|
|
|
|
|
|
_result.create(image.rows - templ.rows + 1, image.cols - templ.cols + 1, CV_32FC1);
|
|
|
|
GpuMat result = _result.getGpuMat();
|
|
|
|
|
2012-10-17 15:12:04 +08:00
|
|
|
matchTemplateNaive_SQDIFF_32F(image, templ, result, image.channels(), StreamAccessor::getStream(stream));
|
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
///////////////////////////////////////////////////////////////
|
|
|
|
// SQDIFF_8U
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
class Match_SQDIFF_8U : public TemplateMatching
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
public:
|
|
|
|
explicit Match_SQDIFF_8U(Size user_block_size) : match_CCORR_(user_block_size)
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void match(InputArray image, InputArray templ, OutputArray result, Stream& stream = Stream::Null());
|
|
|
|
|
|
|
|
private:
|
|
|
|
GpuMat image_sqsums_;
|
|
|
|
GpuMat intBuffer_;
|
|
|
|
Match_CCORR_8U match_CCORR_;
|
|
|
|
};
|
|
|
|
|
|
|
|
void Match_SQDIFF_8U::match(InputArray _image, InputArray _templ, OutputArray _result, Stream& stream)
|
|
|
|
{
|
|
|
|
using namespace cv::gpu::cudev::match_template;
|
|
|
|
|
|
|
|
GpuMat image = _image.getGpuMat();
|
|
|
|
GpuMat templ = _templ.getGpuMat();
|
|
|
|
|
|
|
|
CV_Assert( image.depth() == CV_8U );
|
|
|
|
CV_Assert( image.type() == templ.type() );
|
|
|
|
CV_Assert( image.cols >= templ.cols && image.rows >= templ.rows );
|
|
|
|
|
|
|
|
if (templ.size().area() < getTemplateThreshold(TM_SQDIFF, CV_8U))
|
|
|
|
{
|
|
|
|
_result.create(image.rows - templ.rows + 1, image.cols - templ.cols + 1, CV_32FC1);
|
|
|
|
GpuMat result = _result.getGpuMat();
|
|
|
|
|
2012-10-17 15:12:04 +08:00
|
|
|
matchTemplateNaive_SQDIFF_8U(image, templ, result, image.channels(), StreamAccessor::getStream(stream));
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
gpu::sqrIntegral(image.reshape(1), image_sqsums_, intBuffer_, stream);
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
unsigned long long templ_sqsum = (unsigned long long) gpu::sqrSum(templ.reshape(1))[0];
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
match_CCORR_.match(image, templ, _result, stream);
|
|
|
|
GpuMat result = _result.getGpuMat();
|
|
|
|
|
|
|
|
matchTemplatePrepared_SQDIFF_8U(templ.cols, templ.rows, image_sqsums_, templ_sqsum, result, image.channels(), StreamAccessor::getStream(stream));
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
///////////////////////////////////////////////////////////////
|
|
|
|
// SQDIFF_NORMED_8U
|
|
|
|
|
|
|
|
class Match_SQDIFF_NORMED_8U : public TemplateMatching
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
explicit Match_SQDIFF_NORMED_8U(Size user_block_size) : match_CCORR_(user_block_size)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
void match(InputArray image, InputArray templ, OutputArray result, Stream& stream = Stream::Null());
|
|
|
|
|
|
|
|
private:
|
|
|
|
GpuMat image_sqsums_;
|
|
|
|
GpuMat intBuffer_;
|
|
|
|
Match_CCORR_8U match_CCORR_;
|
|
|
|
};
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
void Match_SQDIFF_NORMED_8U::match(InputArray _image, InputArray _templ, OutputArray _result, Stream& stream)
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
using namespace cv::gpu::cudev::match_template;
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
GpuMat image = _image.getGpuMat();
|
|
|
|
GpuMat templ = _templ.getGpuMat();
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
CV_Assert( image.depth() == CV_8U );
|
|
|
|
CV_Assert( image.type() == templ.type() );
|
|
|
|
CV_Assert( image.cols >= templ.cols && image.rows >= templ.rows );
|
|
|
|
|
|
|
|
gpu::sqrIntegral(image.reshape(1), image_sqsums_, intBuffer_, stream);
|
|
|
|
|
|
|
|
unsigned long long templ_sqsum = (unsigned long long) gpu::sqrSum(templ.reshape(1))[0];
|
|
|
|
|
|
|
|
match_CCORR_.match(image, templ, _result, stream);
|
|
|
|
GpuMat result = _result.getGpuMat();
|
|
|
|
|
|
|
|
matchTemplatePrepared_SQDIFF_NORMED_8U(templ.cols, templ.rows, image_sqsums_, templ_sqsum, result, image.channels(), StreamAccessor::getStream(stream));
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
///////////////////////////////////////////////////////////////
|
|
|
|
// CCOFF_8U
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
class Match_CCOEFF_8U : public TemplateMatching
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
public:
|
|
|
|
explicit Match_CCOEFF_8U(Size user_block_size) : match_CCORR_(user_block_size)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
void match(InputArray image, InputArray templ, OutputArray result, Stream& stream = Stream::Null());
|
|
|
|
|
|
|
|
private:
|
|
|
|
GpuMat intBuffer_;
|
|
|
|
std::vector<GpuMat> images_;
|
|
|
|
std::vector<GpuMat> image_sums_;
|
|
|
|
Match_CCORR_8U match_CCORR_;
|
|
|
|
};
|
|
|
|
|
|
|
|
void Match_CCOEFF_8U::match(InputArray _image, InputArray _templ, OutputArray _result, Stream& stream)
|
|
|
|
{
|
|
|
|
using namespace cv::gpu::cudev::match_template;
|
|
|
|
|
|
|
|
GpuMat image = _image.getGpuMat();
|
|
|
|
GpuMat templ = _templ.getGpuMat();
|
|
|
|
|
|
|
|
CV_Assert( image.depth() == CV_8U );
|
|
|
|
CV_Assert( image.type() == templ.type() );
|
|
|
|
CV_Assert( image.cols >= templ.cols && image.rows >= templ.rows );
|
|
|
|
|
|
|
|
match_CCORR_.match(image, templ, _result, stream);
|
|
|
|
GpuMat result = _result.getGpuMat();
|
2012-10-17 15:12:04 +08:00
|
|
|
|
|
|
|
if (image.channels() == 1)
|
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
image_sums_.resize(1);
|
|
|
|
gpu::integral(image, image_sums_[0], intBuffer_, stream);
|
|
|
|
|
|
|
|
unsigned int templ_sum = (unsigned int) gpu::sum(templ)[0];
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
matchTemplatePrepared_CCOFF_8U(templ.cols, templ.rows, image_sums_[0], templ_sum, result, StreamAccessor::getStream(stream));
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
gpu::split(image, images_);
|
|
|
|
|
|
|
|
image_sums_.resize(images_.size());
|
2012-10-17 15:12:04 +08:00
|
|
|
for (int i = 0; i < image.channels(); ++i)
|
2013-04-30 21:27:06 +08:00
|
|
|
gpu::integral(images_[i], image_sums_[i], intBuffer_, stream);
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-06-04 17:32:35 +08:00
|
|
|
Scalar templ_sum = gpu::sum(templ);
|
2012-10-17 15:12:04 +08:00
|
|
|
|
|
|
|
switch (image.channels())
|
|
|
|
{
|
|
|
|
case 2:
|
|
|
|
matchTemplatePrepared_CCOFF_8UC2(
|
2013-04-30 21:27:06 +08:00
|
|
|
templ.cols, templ.rows, image_sums_[0], image_sums_[1],
|
|
|
|
(unsigned int) templ_sum[0], (unsigned int) templ_sum[1],
|
2012-10-17 15:12:04 +08:00
|
|
|
result, StreamAccessor::getStream(stream));
|
|
|
|
break;
|
|
|
|
case 3:
|
|
|
|
matchTemplatePrepared_CCOFF_8UC3(
|
2013-04-30 21:27:06 +08:00
|
|
|
templ.cols, templ.rows, image_sums_[0], image_sums_[1], image_sums_[2],
|
|
|
|
(unsigned int) templ_sum[0], (unsigned int) templ_sum[1], (unsigned int) templ_sum[2],
|
2012-10-17 15:12:04 +08:00
|
|
|
result, StreamAccessor::getStream(stream));
|
|
|
|
break;
|
|
|
|
case 4:
|
|
|
|
matchTemplatePrepared_CCOFF_8UC4(
|
2013-04-30 21:27:06 +08:00
|
|
|
templ.cols, templ.rows, image_sums_[0], image_sums_[1], image_sums_[2], image_sums_[3],
|
|
|
|
(unsigned int) templ_sum[0], (unsigned int) templ_sum[1], (unsigned int) templ_sum[2], (unsigned int) templ_sum[3],
|
|
|
|
result, StreamAccessor::getStream(stream));
|
2012-10-17 15:12:04 +08:00
|
|
|
break;
|
|
|
|
default:
|
2013-04-30 21:27:06 +08:00
|
|
|
CV_Error(Error::StsBadArg, "unsupported number of channels");
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
///////////////////////////////////////////////////////////////
|
|
|
|
// CCOFF_NORMED_8U
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
class Match_CCOEFF_NORMED_8U : public TemplateMatching
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
public:
|
|
|
|
explicit Match_CCOEFF_NORMED_8U(Size user_block_size) : match_CCORR_32F_(user_block_size)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
void match(InputArray image, InputArray templ, OutputArray result, Stream& stream = Stream::Null());
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
private:
|
|
|
|
GpuMat imagef_, templf_;
|
|
|
|
Match_CCORR_32F match_CCORR_32F_;
|
|
|
|
GpuMat intBuffer_;
|
|
|
|
std::vector<GpuMat> images_;
|
|
|
|
std::vector<GpuMat> image_sums_;
|
|
|
|
std::vector<GpuMat> image_sqsums_;
|
|
|
|
};
|
|
|
|
|
|
|
|
void Match_CCOEFF_NORMED_8U::match(InputArray _image, InputArray _templ, OutputArray _result, Stream& stream)
|
|
|
|
{
|
|
|
|
using namespace cv::gpu::cudev::match_template;
|
|
|
|
|
|
|
|
GpuMat image = _image.getGpuMat();
|
|
|
|
GpuMat templ = _templ.getGpuMat();
|
|
|
|
|
|
|
|
CV_Assert( image.depth() == CV_8U );
|
|
|
|
CV_Assert( image.type() == templ.type() );
|
|
|
|
CV_Assert( image.cols >= templ.cols && image.rows >= templ.rows );
|
|
|
|
|
|
|
|
image.convertTo(imagef_, CV_32F, stream);
|
|
|
|
templ.convertTo(templf_, CV_32F, stream);
|
|
|
|
|
|
|
|
match_CCORR_32F_.match(imagef_, templf_, _result, stream);
|
|
|
|
GpuMat result = _result.getGpuMat();
|
2012-10-17 15:12:04 +08:00
|
|
|
|
|
|
|
if (image.channels() == 1)
|
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
image_sums_.resize(1);
|
|
|
|
gpu::integral(image, image_sums_[0], intBuffer_, stream);
|
|
|
|
|
|
|
|
image_sqsums_.resize(1);
|
|
|
|
gpu::sqrIntegral(image, image_sqsums_[0], intBuffer_, stream);
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
unsigned int templ_sum = (unsigned int) gpu::sum(templ)[0];
|
|
|
|
unsigned long long templ_sqsum = (unsigned long long) gpu::sqrSum(templ)[0];
|
2012-10-17 15:12:04 +08:00
|
|
|
|
|
|
|
matchTemplatePrepared_CCOFF_NORMED_8U(
|
2013-04-30 21:27:06 +08:00
|
|
|
templ.cols, templ.rows, image_sums_[0], image_sqsums_[0],
|
2012-10-17 15:12:04 +08:00
|
|
|
templ_sum, templ_sqsum, result, StreamAccessor::getStream(stream));
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
gpu::split(image, images_);
|
|
|
|
|
|
|
|
image_sums_.resize(images_.size());
|
|
|
|
image_sqsums_.resize(images_.size());
|
2012-10-17 15:12:04 +08:00
|
|
|
for (int i = 0; i < image.channels(); ++i)
|
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
gpu::integral(images_[i], image_sums_[i], intBuffer_, stream);
|
|
|
|
gpu::sqrIntegral(images_[i], image_sqsums_[i], intBuffer_, stream);
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
|
|
|
|
2013-06-04 17:32:35 +08:00
|
|
|
Scalar templ_sum = gpu::sum(templ);
|
|
|
|
Scalar templ_sqsum = gpu::sqrSum(templ);
|
2012-10-17 15:12:04 +08:00
|
|
|
|
|
|
|
switch (image.channels())
|
|
|
|
{
|
|
|
|
case 2:
|
|
|
|
matchTemplatePrepared_CCOFF_NORMED_8UC2(
|
|
|
|
templ.cols, templ.rows,
|
2013-04-30 21:27:06 +08:00
|
|
|
image_sums_[0], image_sqsums_[0],
|
|
|
|
image_sums_[1], image_sqsums_[1],
|
2012-10-17 15:12:04 +08:00
|
|
|
(unsigned int)templ_sum[0], (unsigned long long)templ_sqsum[0],
|
|
|
|
(unsigned int)templ_sum[1], (unsigned long long)templ_sqsum[1],
|
|
|
|
result, StreamAccessor::getStream(stream));
|
|
|
|
break;
|
|
|
|
case 3:
|
|
|
|
matchTemplatePrepared_CCOFF_NORMED_8UC3(
|
|
|
|
templ.cols, templ.rows,
|
2013-04-30 21:27:06 +08:00
|
|
|
image_sums_[0], image_sqsums_[0],
|
|
|
|
image_sums_[1], image_sqsums_[1],
|
|
|
|
image_sums_[2], image_sqsums_[2],
|
2012-10-17 15:12:04 +08:00
|
|
|
(unsigned int)templ_sum[0], (unsigned long long)templ_sqsum[0],
|
|
|
|
(unsigned int)templ_sum[1], (unsigned long long)templ_sqsum[1],
|
|
|
|
(unsigned int)templ_sum[2], (unsigned long long)templ_sqsum[2],
|
|
|
|
result, StreamAccessor::getStream(stream));
|
|
|
|
break;
|
|
|
|
case 4:
|
|
|
|
matchTemplatePrepared_CCOFF_NORMED_8UC4(
|
|
|
|
templ.cols, templ.rows,
|
2013-04-30 21:27:06 +08:00
|
|
|
image_sums_[0], image_sqsums_[0],
|
|
|
|
image_sums_[1], image_sqsums_[1],
|
|
|
|
image_sums_[2], image_sqsums_[2],
|
|
|
|
image_sums_[3], image_sqsums_[3],
|
2012-10-17 15:12:04 +08:00
|
|
|
(unsigned int)templ_sum[0], (unsigned long long)templ_sqsum[0],
|
|
|
|
(unsigned int)templ_sum[1], (unsigned long long)templ_sqsum[1],
|
|
|
|
(unsigned int)templ_sum[2], (unsigned long long)templ_sqsum[2],
|
|
|
|
(unsigned int)templ_sum[3], (unsigned long long)templ_sqsum[3],
|
|
|
|
result, StreamAccessor::getStream(stream));
|
|
|
|
break;
|
|
|
|
default:
|
2013-04-30 21:27:06 +08:00
|
|
|
CV_Error(Error::StsBadArg, "unsupported number of channels");
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
Ptr<gpu::TemplateMatching> cv::gpu::createTemplateMatching(int srcType, int method, Size user_block_size)
|
2012-10-17 15:12:04 +08:00
|
|
|
{
|
2013-04-30 21:27:06 +08:00
|
|
|
const int sdepth = CV_MAT_DEPTH(srcType);
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
CV_Assert( sdepth == CV_8U || sdepth == CV_32F );
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
if (sdepth == CV_32F)
|
|
|
|
{
|
|
|
|
switch (method)
|
|
|
|
{
|
|
|
|
case TM_SQDIFF:
|
|
|
|
return new Match_SQDIFF_32F;
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
case TM_CCORR:
|
|
|
|
return new Match_CCORR_32F(user_block_size);
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
default:
|
|
|
|
CV_Error( Error::StsBadFlag, "Unsopported method" );
|
|
|
|
return Ptr<gpu::TemplateMatching>();
|
|
|
|
}
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
2013-04-30 21:27:06 +08:00
|
|
|
else
|
|
|
|
{
|
|
|
|
switch (method)
|
|
|
|
{
|
|
|
|
case TM_SQDIFF:
|
|
|
|
return new Match_SQDIFF_8U(user_block_size);
|
|
|
|
|
|
|
|
case TM_SQDIFF_NORMED:
|
|
|
|
return new Match_SQDIFF_NORMED_8U(user_block_size);
|
2012-10-17 15:12:04 +08:00
|
|
|
|
2013-04-30 21:27:06 +08:00
|
|
|
case TM_CCORR:
|
|
|
|
return new Match_CCORR_8U(user_block_size);
|
|
|
|
|
|
|
|
case TM_CCORR_NORMED:
|
|
|
|
return new Match_CCORR_NORMED_8U(user_block_size);
|
|
|
|
|
|
|
|
case TM_CCOEFF:
|
|
|
|
return new Match_CCOEFF_8U(user_block_size);
|
|
|
|
|
|
|
|
case TM_CCOEFF_NORMED:
|
|
|
|
return new Match_CCOEFF_NORMED_8U(user_block_size);
|
|
|
|
|
|
|
|
default:
|
|
|
|
CV_Error( Error::StsBadFlag, "Unsopported method" );
|
|
|
|
return Ptr<gpu::TemplateMatching>();
|
|
|
|
}
|
|
|
|
}
|
2012-10-17 15:12:04 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|