mirror of
https://github.com/opencv/opencv.git
synced 2024-12-04 16:59:12 +08:00
0f622206e4
* integrated the new C++ persistence; removed old persistence; most of OpenCV compiles fine! the tests have not been run yet * fixed multiple bugs in the new C++ persistence * fixed raw size of the parsed empty sequences * [temporarily] excluded obsolete applications traincascade and createsamples from build * fixed several compiler warnings and multiple test failures * undo changes in cocoa window rendering (that was fixed in another PR) * fixed more compile warnings and the remaining test failures (hopefully) * trying to fix the last little warning
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( const cv::FileStorage& 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
|