diff options
Diffstat (limited to 'thirdparty1/linux/include/opencv2/face/predict_collector.hpp')
-rw-r--r-- | thirdparty1/linux/include/opencv2/face/predict_collector.hpp | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/thirdparty1/linux/include/opencv2/face/predict_collector.hpp b/thirdparty1/linux/include/opencv2/face/predict_collector.hpp new file mode 100644 index 0000000..a9f907d --- /dev/null +++ b/thirdparty1/linux/include/opencv2/face/predict_collector.hpp @@ -0,0 +1,127 @@ +/* +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 + (3-clause BSD License) + +Copyright (C) 2000-2015, Intel Corporation, all rights reserved. +Copyright (C) 2009-2011, Willow Garage Inc., all rights reserved. +Copyright (C) 2009-2015, NVIDIA Corporation, all rights reserved. +Copyright (C) 2010-2013, Advanced Micro Devices, Inc., all rights reserved. +Copyright (C) 2015, OpenCV Foundation, all rights reserved. +Copyright (C) 2015, Itseez 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: + + * Redistributions of source code must retain the above copyright notice, + this list of conditions and the following disclaimer. + + * Redistributions 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. + + * Neither the names of the copyright holders nor the names of the contributors + may 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 copyright holders 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. +*/ + +#ifndef __OPENCV_PREDICT_COLLECTOR_HPP__ +#define __OPENCV_PREDICT_COLLECTOR_HPP__ + +#include <vector> +#include <map> +#include <utility> +#include <cfloat> + +#include "opencv2/core/cvstd.hpp" + +namespace cv { +namespace face { +//! @addtogroup face +//! @{ +/** @brief Abstract base class for all strategies of prediction result handling +*/ +class CV_EXPORTS_W PredictCollector +{ +public: + virtual ~PredictCollector() {} + + /** @brief Interface method called by face recognizer before results processing + @param size total size of prediction evaluation that recognizer could perform + */ + virtual void init(size_t size) { (void)size; } + + /** @brief Interface method called by face recognizer for each result + @param label current prediction label + @param dist current prediction distance (confidence) + */ + virtual bool collect(int label, double dist) = 0; +}; + +/** @brief Default predict collector + +Trace minimal distance with treshhold checking (that is default behavior for most predict logic) +*/ +class CV_EXPORTS_W StandardCollector : public PredictCollector +{ +public: + struct PredictResult + { + int label; + double distance; + PredictResult(int label_ = -1, double distance_ = DBL_MAX) : label(label_), distance(distance_) {} + }; +protected: + double threshold; + PredictResult minRes; + std::vector<PredictResult> data; +public: + /** @brief Constructor + @param threshold_ set threshold + */ + StandardCollector(double threshold_ = DBL_MAX); + /** @brief overloaded interface method */ + void init(size_t size); + /** @brief overloaded interface method */ + bool collect(int label, double dist); + /** @brief Returns label with minimal distance */ + CV_WRAP int getMinLabel() const; + /** @brief Returns minimal distance value */ + CV_WRAP double getMinDist() const; + /** @brief Return results as vector + @param sorted If set, results will be sorted by distance + Each values is a pair of label and distance. + */ + CV_WRAP std::vector< std::pair<int, double> > getResults(bool sorted = false) const; + /** @brief Return results as map + Labels are keys, values are minimal distances + */ + std::map<int, double> getResultsMap() const; + /** @brief Static constructor + @param threshold set threshold + */ + CV_WRAP static Ptr<StandardCollector> create(double threshold = DBL_MAX); +}; + +//! @} +} +} + +#endif |