mirror of
https://github.com/opencv/opencv.git
synced 2024-12-25 18:18:04 +08:00
d86d8ed909
PR #2968:cce2d998578f9cFixed bug which caused crash of GPU version of feature matcher in stitcher The bug caused crash of GPU version of feature matcher in stitcher when we use ORB features. PR #3236:5947519Check sure that we're not already below required leaf false alarm rate before continuing to get negative samples. PR #3190 fix blobdetector PR #3562 (part):82bd82eTBB updated to 4.3u2. Fix for aarch64 support. PR #3604 (part):091c7a3OpenGL interop sample reworked not ot use cvconfig.h PR #3792:afdf319Add -L for CUDA libs path to pkg-config Add all dirs from CUDA_LIBS_PATH as -L linker options to OPENCV_LINKER_LIBS. These will end up in opencv.pc. PR #3893:122b9f8Turn ocv_convert_to_lib_name into a function PR #5490:ec5244afixed memory leak in findHomography tests PR #5491:0d5b739delete video readers PR #5574 PR #5202
126 lines
3.9 KiB
C++
126 lines
3.9 KiB
C++
#ifndef _OPENCV_CASCADECLASSIFIER_H_
|
|
#define _OPENCV_CASCADECLASSIFIER_H_
|
|
|
|
#include <ctime>
|
|
#include "traincascade_features.h"
|
|
#include "haarfeatures.h"
|
|
#include "lbpfeatures.h"
|
|
#include "HOGfeatures.h" //new
|
|
#include "boost.h"
|
|
|
|
#define CC_CASCADE_FILENAME "cascade.xml"
|
|
#define CC_PARAMS_FILENAME "params.xml"
|
|
|
|
#define CC_CASCADE_PARAMS "cascadeParams"
|
|
#define CC_STAGE_TYPE "stageType"
|
|
#define CC_FEATURE_TYPE "featureType"
|
|
#define CC_HEIGHT "height"
|
|
#define CC_WIDTH "width"
|
|
|
|
#define CC_STAGE_NUM "stageNum"
|
|
#define CC_STAGES "stages"
|
|
#define CC_STAGE_PARAMS "stageParams"
|
|
|
|
#define CC_BOOST "BOOST"
|
|
#define CC_BOOST_TYPE "boostType"
|
|
#define CC_DISCRETE_BOOST "DAB"
|
|
#define CC_REAL_BOOST "RAB"
|
|
#define CC_LOGIT_BOOST "LB"
|
|
#define CC_GENTLE_BOOST "GAB"
|
|
#define CC_MINHITRATE "minHitRate"
|
|
#define CC_MAXFALSEALARM "maxFalseAlarm"
|
|
#define CC_TRIM_RATE "weightTrimRate"
|
|
#define CC_MAX_DEPTH "maxDepth"
|
|
#define CC_WEAK_COUNT "maxWeakCount"
|
|
#define CC_STAGE_THRESHOLD "stageThreshold"
|
|
#define CC_WEAK_CLASSIFIERS "weakClassifiers"
|
|
#define CC_INTERNAL_NODES "internalNodes"
|
|
#define CC_LEAF_VALUES "leafValues"
|
|
|
|
#define CC_FEATURES FEATURES
|
|
#define CC_FEATURE_PARAMS "featureParams"
|
|
#define CC_MAX_CAT_COUNT "maxCatCount"
|
|
#define CC_FEATURE_SIZE "featSize"
|
|
|
|
#define CC_HAAR "HAAR"
|
|
#define CC_MODE "mode"
|
|
#define CC_MODE_BASIC "BASIC"
|
|
#define CC_MODE_CORE "CORE"
|
|
#define CC_MODE_ALL "ALL"
|
|
#define CC_RECTS "rects"
|
|
#define CC_TILTED "tilted"
|
|
|
|
#define CC_LBP "LBP"
|
|
#define CC_RECT "rect"
|
|
|
|
#define CC_HOG "HOG"
|
|
|
|
#ifdef _WIN32
|
|
#define TIME( arg ) (((double) clock()) / CLOCKS_PER_SEC)
|
|
#else
|
|
#define TIME( arg ) (time( arg ))
|
|
#endif
|
|
|
|
class CvCascadeParams : public CvParams
|
|
{
|
|
public:
|
|
enum { BOOST = 0 };
|
|
static const int defaultStageType = BOOST;
|
|
static const int defaultFeatureType = CvFeatureParams::HAAR;
|
|
|
|
CvCascadeParams();
|
|
CvCascadeParams( int _stageType, int _featureType );
|
|
void write( cv::FileStorage &fs ) const;
|
|
bool read( const cv::FileNode &node );
|
|
|
|
void printDefaults() const;
|
|
void printAttrs() const;
|
|
bool scanAttr( const std::string prmName, const std::string val );
|
|
|
|
int stageType;
|
|
int featureType;
|
|
cv::Size winSize;
|
|
};
|
|
|
|
class CvCascadeClassifier
|
|
{
|
|
public:
|
|
bool train( const std::string _cascadeDirName,
|
|
const std::string _posFilename,
|
|
const std::string _negFilename,
|
|
int _numPos, int _numNeg,
|
|
int _precalcValBufSize, int _precalcIdxBufSize,
|
|
int _numStages,
|
|
const CvCascadeParams& _cascadeParams,
|
|
const CvFeatureParams& _featureParams,
|
|
const CvCascadeBoostParams& _stageParams,
|
|
bool baseFormatSave = false,
|
|
double acceptanceRatioBreakValue = -1.0 );
|
|
private:
|
|
int predict( int sampleIdx );
|
|
void save( const std::string cascadeDirName, bool baseFormat = false );
|
|
bool load( const std::string cascadeDirName );
|
|
bool updateTrainingSet( double minimumAcceptanceRatio, double& acceptanceRatio );
|
|
int fillPassedSamples( int first, int count, bool isPositive, double requiredAcceptanceRatio, int64& consumed );
|
|
|
|
void writeParams( cv::FileStorage &fs ) const;
|
|
void writeStages( cv::FileStorage &fs, const cv::Mat& featureMap ) const;
|
|
void writeFeatures( cv::FileStorage &fs, const cv::Mat& featureMap ) const;
|
|
bool readParams( const cv::FileNode &node );
|
|
bool readStages( const cv::FileNode &node );
|
|
|
|
void getUsedFeaturesIdxMap( cv::Mat& featureMap );
|
|
|
|
CvCascadeParams cascadeParams;
|
|
cv::Ptr<CvFeatureParams> featureParams;
|
|
cv::Ptr<CvCascadeBoostParams> stageParams;
|
|
|
|
cv::Ptr<CvFeatureEvaluator> featureEvaluator;
|
|
std::vector< cv::Ptr<CvCascadeBoost> > stageClassifiers;
|
|
CvCascadeImageReader imgReader;
|
|
int numStages, curNumSamples;
|
|
int numPos, numNeg;
|
|
};
|
|
|
|
#endif
|