mirror of
https://github.com/opencv/opencv.git
synced 2024-11-26 04:00:30 +08:00
4a297a2443
- removed tr1 usage (dropped in C++17) - moved includes of vector/map/iostream/limits into ts.hpp - require opencv_test + anonymous namespace (added compile check) - fixed norm() usage (must be from cvtest::norm for checks) and other conflict functions - added missing license headers
93 lines
2.3 KiB
C++
93 lines
2.3 KiB
C++
#ifndef __OPENCV_TEST_PRECOMP_HPP__
|
|
#define __OPENCV_TEST_PRECOMP_HPP__
|
|
|
|
#include "opencv2/ts.hpp"
|
|
#include "opencv2/ml.hpp"
|
|
#include "opencv2/core/core_c.h"
|
|
|
|
namespace opencv_test {
|
|
using namespace cv::ml;
|
|
|
|
#define CV_NBAYES "nbayes"
|
|
#define CV_KNEAREST "knearest"
|
|
#define CV_SVM "svm"
|
|
#define CV_EM "em"
|
|
#define CV_ANN "ann"
|
|
#define CV_DTREE "dtree"
|
|
#define CV_BOOST "boost"
|
|
#define CV_RTREES "rtrees"
|
|
#define CV_ERTREES "ertrees"
|
|
#define CV_SVMSGD "svmsgd"
|
|
|
|
enum { CV_TRAIN_ERROR=0, CV_TEST_ERROR=1 };
|
|
|
|
using cv::Ptr;
|
|
using cv::ml::StatModel;
|
|
using cv::ml::TrainData;
|
|
using cv::ml::NormalBayesClassifier;
|
|
using cv::ml::SVM;
|
|
using cv::ml::KNearest;
|
|
using cv::ml::ParamGrid;
|
|
using cv::ml::ANN_MLP;
|
|
using cv::ml::DTrees;
|
|
using cv::ml::Boost;
|
|
using cv::ml::RTrees;
|
|
using cv::ml::SVMSGD;
|
|
|
|
class CV_MLBaseTest : public cvtest::BaseTest
|
|
{
|
|
public:
|
|
CV_MLBaseTest( const char* _modelName );
|
|
virtual ~CV_MLBaseTest();
|
|
protected:
|
|
virtual int read_params( CvFileStorage* fs );
|
|
virtual void run( int startFrom );
|
|
virtual int prepare_test_case( int testCaseIdx );
|
|
virtual std::string& get_validation_filename();
|
|
virtual int run_test_case( int testCaseIdx ) = 0;
|
|
virtual int validate_test_results( int testCaseIdx ) = 0;
|
|
|
|
int train( int testCaseIdx );
|
|
float get_test_error( int testCaseIdx, std::vector<float> *resp = 0 );
|
|
void save( const char* filename );
|
|
void load( const char* filename );
|
|
|
|
Ptr<TrainData> data;
|
|
std::string modelName, validationFN;
|
|
std::vector<std::string> dataSetNames;
|
|
cv::FileStorage validationFS;
|
|
|
|
Ptr<StatModel> model;
|
|
|
|
std::map<int, int> cls_map;
|
|
|
|
int64 initSeed;
|
|
};
|
|
|
|
class CV_AMLTest : public CV_MLBaseTest
|
|
{
|
|
public:
|
|
CV_AMLTest( const char* _modelName );
|
|
virtual ~CV_AMLTest() {}
|
|
protected:
|
|
virtual int run_test_case( int testCaseIdx );
|
|
virtual int validate_test_results( int testCaseIdx );
|
|
};
|
|
|
|
class CV_SLMLTest : public CV_MLBaseTest
|
|
{
|
|
public:
|
|
CV_SLMLTest( const char* _modelName );
|
|
virtual ~CV_SLMLTest() {}
|
|
protected:
|
|
virtual int run_test_case( int testCaseIdx );
|
|
virtual int validate_test_results( int testCaseIdx );
|
|
|
|
std::vector<float> test_resps1, test_resps2; // predicted responses for test data
|
|
std::string fname1, fname2;
|
|
};
|
|
|
|
} // namespace
|
|
|
|
#endif
|