From: marina.kolpakova Date: Wed, 19 Sep 2012 23:28:45 +0000 (+0400) Subject: GPU interface for soft cascade X-Git-Tag: submit/tizen_ivi/20141117.190038~2^2~1214^2~59^2~72 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=dd9c53497bb8c32d411d8b5e784b249ad9d13364;p=profile%2Fivi%2Fopencv.git GPU interface for soft cascade --- diff --git a/modules/gpu/include/opencv2/gpu/gpu.hpp b/modules/gpu/include/opencv2/gpu/gpu.hpp index ddb1317..4a2d88a 100644 --- a/modules/gpu/include/opencv2/gpu/gpu.hpp +++ b/modules/gpu/include/opencv2/gpu/gpu.hpp @@ -1532,6 +1532,49 @@ public: int detectMultiScale(const GpuMat& image, GpuMat& objectsBuf, Size maxObjectSize, Size minSize = Size(), double scaleFactor = 1.1, int minNeighbors = 4); }; +// ======================== GPU version for soft cascade ===================== // + +class CV_EXPORTS SoftCascade +{ +public: + //! An empty cascade will be created. + SoftCascade(); + + //! Cascade will be created from file for scales from minScale to maxScale. + //! Param filename is a path to xml-serialized cascade. + //! Param minScale is a minimum scale relative to the original size of the image on which cascade will be applyed. + //! Param minScale is a maximum scale relative to the original size of the image on which cascade will be applyed. + SoftCascade( const string& filename, const float minScale = 0.4f, const float maxScale = 5.f); + + //! cascade will be loaded from file "filename". The previous cascade will be destroyed. + //! Param filename is a path to xml-serialized cascade. + //! Param minScale is a minimum scale relative to the original size of the image on which cascade will be applyed. + //! Param minScale is a maximum scale relative to the original size of the image on which cascade will be applyed. + bool load( const string& filename, const float minScale = 0.4f, const float maxScale = 5.f); + + virtual ~SoftCascade(); + + //! return vector of bounding boxes. Each box contains one detected object + virtual void detectMultiScale(const GpuMat& image, const GpuMat& rois, GpuMat& objects, + int rejectfactor = 1, Stream stream = Stream::Null()); // ToDo store objects in GPU mem + +protected: + enum { BOOST = 0 }; + enum + { + FRAME_WIDTH = 640, + FRAME_HEIGHT = 480, + TOTAL_SCALES = 55, + CLASSIFIERS = 5, + ORIG_OBJECT_WIDTH = 64, + ORIG_OBJECT_HEIGHT = 128 + }; + +private: + struct Filds; + Filds* filds; +}; + ////////////////////////////////// SURF ////////////////////////////////////////// class CV_EXPORTS SURF_GPU diff --git a/modules/gpu/src/softcascade.cpp b/modules/gpu/src/softcascade.cpp new file mode 100644 index 0000000..509e3f5 --- /dev/null +++ b/modules/gpu/src/softcascade.cpp @@ -0,0 +1,100 @@ +/*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) 2008-2012, 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 + +#if !defined (HAVE_CUDA) + +cv::gpu::SoftCascade::SoftCascade() : filds(0) { throw_nogpu(); } + +cv::gpu::SoftCascade::SoftCascade( const string&, const float, const float) : filds(0) { throw_nogpu(); } + +cv::gpu::SoftCascade::~SoftCascade() { throw_nogpu(); } + +bool cv::gpu::SoftCascade::load( const string&, const float, const float) { throw_nogpu(); } + +void cv::gpu::SoftCascade::detectMultiScale(const GpuMat&, const GpuMat&, GpuMat&, const int, Stream) { throw_nogpu(); } + +#else + +struct cv::gpu::SoftCascade::Filds +{ + bool fill(const FileNode &root, const float mins, const float maxs){return true;} + void calcLevels(int frameW, int frameH, int scales) {} +}; + +cv::gpu::SoftCascade::SoftCascade() : filds(0) {} + +cv::gpu::SoftCascade::SoftCascade( const string& filename, const float minScale, const float maxScale) : filds(0) +{ + load(filename, minScale, maxScale); +} + +cv::gpu::SoftCascade::~SoftCascade() +{ + delete filds; +} + +bool cv::gpu::SoftCascade::load( const string& filename, const float minScale, const float maxScale) +{ + if (filds) + delete filds; + filds = 0; + + cv::FileStorage fs(filename, FileStorage::READ); + if (!fs.isOpened()) return false; + + filds = new Filds; + Filds& flds = *filds; + if (!flds.fill(fs.getFirstTopLevelNode(), minScale, maxScale)) return false; + flds.calcLevels(FRAME_WIDTH, FRAME_HEIGHT, TOTAL_SCALES); + + return true; +} + +void cv::gpu::SoftCascade::detectMultiScale(const GpuMat& /*image*/, const GpuMat& /*rois*/, + GpuMat& /*objects*/, const int /*rejectfactor*/, Stream /*stream*/) +{ + // empty +} + +#endif