Vladislav Vinogradov
10 years ago
parent
b2a5e66fc5
commit
4cc7b0a74f
7 changed files with 161 additions and 163 deletions
@ -0,0 +1,147 @@ |
||||
/*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::cuda; |
||||
|
||||
#if !defined HAVE_CUDA || defined(CUDA_DISABLER) |
||||
|
||||
Ptr<ImagePyramid> cv::cuda::createImagePyramid(InputArray, int, Stream&) { throw_no_cuda(); return Ptr<ImagePyramid>(); } |
||||
|
||||
#else // HAVE_CUDA
|
||||
|
||||
namespace |
||||
{ |
||||
class ImagePyramidImpl : public ImagePyramid |
||||
{ |
||||
public: |
||||
ImagePyramidImpl(InputArray img, int nLayers, Stream& stream); |
||||
|
||||
void getLayer(OutputArray outImg, Size outRoi, Stream& stream = Stream::Null()) const; |
||||
|
||||
private: |
||||
GpuMat layer0_; |
||||
std::vector<GpuMat> pyramid_; |
||||
int nLayers_; |
||||
}; |
||||
|
||||
ImagePyramidImpl::ImagePyramidImpl(InputArray _img, int numLayers, Stream& stream) |
||||
{ |
||||
GpuMat img = _img.getGpuMat(); |
||||
|
||||
CV_Assert( img.depth() <= CV_32F && img.channels() <= 4 ); |
||||
|
||||
img.copyTo(layer0_, stream); |
||||
|
||||
Size szLastLayer = img.size(); |
||||
nLayers_ = 1; |
||||
|
||||
if (numLayers <= 0) |
||||
numLayers = 255; // it will cut-off when any of the dimensions goes 1
|
||||
|
||||
pyramid_.resize(numLayers); |
||||
|
||||
for (int i = 0; i < numLayers - 1; ++i) |
||||
{ |
||||
Size szCurLayer(szLastLayer.width / 2, szLastLayer.height / 2); |
||||
|
||||
if (szCurLayer.width == 0 || szCurLayer.height == 0) |
||||
break; |
||||
|
||||
ensureSizeIsEnough(szCurLayer, img.type(), pyramid_[i]); |
||||
nLayers_++; |
||||
|
||||
const GpuMat& prevLayer = i == 0 ? layer0_ : pyramid_[i - 1]; |
||||
|
||||
cv::cuda::device::pyramid::downsampleX2(prevLayer, pyramid_[i], img.depth(), img.channels(), StreamAccessor::getStream(stream)); |
||||
|
||||
szLastLayer = szCurLayer; |
||||
} |
||||
} |
||||
|
||||
void ImagePyramidImpl::getLayer(OutputArray _outImg, Size outRoi, Stream& stream) const |
||||
{ |
||||
CV_Assert( outRoi.width <= layer0_.cols && outRoi.height <= layer0_.rows && outRoi.width > 0 && outRoi.height > 0 ); |
||||
|
||||
ensureSizeIsEnough(outRoi, layer0_.type(), _outImg); |
||||
GpuMat outImg = _outImg.getGpuMat(); |
||||
|
||||
if (outRoi.width == layer0_.cols && outRoi.height == layer0_.rows) |
||||
{ |
||||
layer0_.copyTo(outImg, stream); |
||||
return; |
||||
} |
||||
|
||||
float lastScale = 1.0f; |
||||
float curScale; |
||||
GpuMat lastLayer = layer0_; |
||||
GpuMat curLayer; |
||||
|
||||
for (int i = 0; i < nLayers_ - 1; ++i) |
||||
{ |
||||
curScale = lastScale * 0.5f; |
||||
curLayer = pyramid_[i]; |
||||
|
||||
if (outRoi.width == curLayer.cols && outRoi.height == curLayer.rows) |
||||
{ |
||||
curLayer.copyTo(outImg, stream); |
||||
} |
||||
|
||||
if (outRoi.width >= curLayer.cols && outRoi.height >= curLayer.rows) |
||||
break; |
||||
|
||||
lastScale = curScale; |
||||
lastLayer = curLayer; |
||||
} |
||||
|
||||
cv::cuda::device::pyramid::interpolateFrom1(lastLayer, outImg, outImg.depth(), outImg.channels(), StreamAccessor::getStream(stream)); |
||||
} |
||||
} |
||||
|
||||
Ptr<ImagePyramid> cv::cuda::createImagePyramid(InputArray img, int nLayers, Stream& stream) |
||||
{ |
||||
return Ptr<ImagePyramid>(new ImagePyramidImpl(img, nLayers, stream)); |
||||
} |
||||
|
||||
#endif |
||||
Loading…
Reference in new issue