generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

uprated detector test

Browse Source
This commit is contained in:
Maria Dimashova 2010-06-12 16:15:01 +00:00
parent dd72f2ec52
commit d264907fd6
1 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tests/cv/src/adetectordescriptor_evaluation.cpp
View File

@ -535,6 +535,7 @@ protected:
virtual int processResults( int datasetIdx, int caseIdx ) = 0; virtual int processResults( int datasetIdx, int caseIdx ) = 0;
void writeAllPlotData() const; void writeAllPlotData() const;
virtual void writePlotData( int datasetIdx ) const {}; virtual void writePlotData( int datasetIdx ) const {};
virtual void writeAveragePlotData() const {};
string algName; string algName;
bool isWriteParams, isWriteResults, isWriteGraphicsData; bool isWriteParams, isWriteResults, isWriteGraphicsData;
@ -731,6 +732,7 @@ void BaseQualityTest::writeAllPlotData() const
{ {
writePlotData( di ); writePlotData( di );
} }
writeAveragePlotData();
} }
void BaseQualityTest::run ( int ) void BaseQualityTest::run ( int )
@ -817,6 +819,7 @@ protected:
virtual void writeDefaultRunParams( FileStorage &fs ) const; virtual void writeDefaultRunParams( FileStorage &fs ) const;
virtual void writePlotData( int di ) const; virtual void writePlotData( int di ) const;
virtual void writeAveragePlotData() const;
void openToWriteKeypointsFile( FileStorage& fs, int datasetIdx ); void openToWriteKeypointsFile( FileStorage& fs, int datasetIdx );
@ -958,7 +961,6 @@ void DetectorQualityTest::writePlotData(int di ) const
stringstream rFilename, cFilename; stringstream rFilename, cFilename;
rFilename << getPlotPath() << algName << "_" << DATASET_NAMES[di] << "_repeatability.csv"; rFilename << getPlotPath() << algName << "_" << DATASET_NAMES[di] << "_repeatability.csv";
cFilename << getPlotPath() << algName << "_" << DATASET_NAMES[di] << "_correspondenceCount.csv"; cFilename << getPlotPath() << algName << "_" << DATASET_NAMES[di] << "_correspondenceCount.csv";
ofstream rfile(rFilename.str().c_str()), cfile(cFilename.str().c_str()); ofstream rfile(rFilename.str().c_str()), cfile(cFilename.str().c_str());
for( int ci = 0; ci < TEST_CASE_COUNT; ci++ ) for( int ci = 0; ci < TEST_CASE_COUNT; ci++ )
{ {
@ -967,6 +969,27 @@ void DetectorQualityTest::writePlotData(int di ) const
} }
} }
void DetectorQualityTest::writeAveragePlotData() const
{
stringstream rFilename, cFilename;
rFilename << getPlotPath() << algName << "_average_repeatability.csv";
cFilename << getPlotPath() << algName << "_average_correspondenceCount.csv";
ofstream rfile(rFilename.str().c_str()), cfile(cFilename.str().c_str());
float avRep = 0, avCorCount = 0;
for( int di = 0; di < DATASETS_COUNT; di++ )
{
for( int ci = 0; ci < TEST_CASE_COUNT; ci++ )
{
avRep += calcQuality[di][ci].repeatability;
avCorCount += calcQuality[di][ci].correspondenceCount;
}
}
avRep /= DATASETS_COUNT*TEST_CASE_COUNT;
avCorCount /= DATASETS_COUNT*TEST_CASE_COUNT;
rfile << algName << ", " << avRep << endl;
cfile << algName << ", " << cvRound(avCorCount) << endl;
}
void DetectorQualityTest::openToWriteKeypointsFile( FileStorage& fs, int datasetIdx ) void DetectorQualityTest::openToWriteKeypointsFile( FileStorage& fs, int datasetIdx )
{ {
string filename = string(ts->get_data_path()) + KEYPOINTS_DIR + algName + "_"+ string filename = string(ts->get_data_path()) + KEYPOINTS_DIR + algName + "_"+