Commit ccc72f61 authored by Ilya Lysenkov's avatar Ilya Lysenkov

Updated a recall curve construction in descriptors evaluation

parent 13b535ac
...@@ -987,32 +987,26 @@ void DescriptorQualityTest::calculatePlotData( vector<vector<DMatch> > &allMatch ...@@ -987,32 +987,26 @@ void DescriptorQualityTest::calculatePlotData( vector<vector<DMatch> > &allMatch
{ {
vector<Point2f> recallPrecisionCurve; vector<Point2f> recallPrecisionCurve;
computeRecallPrecisionCurve( allMatches, allCorrectMatchesMask, recallPrecisionCurve ); computeRecallPrecisionCurve( allMatches, allCorrectMatchesMask, recallPrecisionCurve );
// you have recallPrecisionCurve for all images from dataset
// size of recallPrecisionCurve == total matches count
#if 0
std::sort( allMatches.begin(), allMatches.end() );
//calcDatasetQuality[di].resize( allMatches.size() );
calcDatasetQuality[di].clear(); calcDatasetQuality[di].clear();
int correctMatchCount = 0, falseMatchCount = 0;
const float sparsePlotBound = 0.1;
const int npoints = 10000;
int step = 1 + allMatches.size() / npoints;
const float resultPrecision = 0.5; const float resultPrecision = 0.5;
bool isResultCalculated = false; bool isResultCalculated = false;
const double eps = 1e-2;
for( size_t i=0;i<allMatches.size();i++) Quality initQuality;
{ initQuality.recall = 0;
if( allMatches[i].isCorrect ) initQuality.precision = 0;
correctMatchCount++; calcDatasetQuality[di].push_back( initQuality );
else
falseMatchCount++;
if( precision( correctMatchCount, falseMatchCount ) >= sparsePlotBound || (i % step == 0) ) for( size_t i=0;i<recallPrecisionCurve.size();i++ )
{ {
Quality quality; Quality quality;
quality.recall = recall( correctMatchCount, allCorrespCount ); quality.recall = recallPrecisionCurve[i].y;
quality.precision = precision( correctMatchCount, falseMatchCount ); quality.precision = 1 - recallPrecisionCurve[i].x;
Quality back = calcDatasetQuality[di].back();
if( fabs( quality.recall - back.recall ) < eps && fabs( quality.precision - back.precision ) < eps )
continue;
calcDatasetQuality[di].push_back( quality ); calcDatasetQuality[di].push_back( quality );
...@@ -1026,15 +1020,6 @@ void DescriptorQualityTest::calculatePlotData( vector<vector<DMatch> > &allMatch ...@@ -1026,15 +1020,6 @@ void DescriptorQualityTest::calculatePlotData( vector<vector<DMatch> > &allMatch
isResultCalculated = true; isResultCalculated = true;
} }
} }
}
Quality quality;
quality.recall = recall( correctMatchCount, allCorrespCount );
quality.precision = precision( correctMatchCount, falseMatchCount );
calcDatasetQuality[di].push_back( quality );
#endif
} }
void DescriptorQualityTest::runDatasetTest (const vector<Mat> &imgs, const vector<Mat> &Hs, int di, int &progress) void DescriptorQualityTest::runDatasetTest (const vector<Mat> &imgs, const vector<Mat> &Hs, int di, int &progress)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment