Fixed gnebehay#11. Added confidence calculation based on number of fu…

…sed points.
glopesdev · Nov 21, 2015 · 38519c9 · 38519c9
1 parent 0cd359d
commit 38519c9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/CMT.cpp b/CMT.cpp
@@ -105,6 +105,9 @@ void CMT::initialize(const Mat im_gray, const Rect rect)
         classes_active = classes_fg;
     }
 
+	//Store initial set of keypoints for confidence calculation
+	points_initial = points_active;
+
     FILE_LOG(logDEBUG) << "CMT::initialize() return";
 }
 
@@ -192,6 +195,9 @@ void CMT::processFrame(Mat im_gray) {
     //TODO: Use theta to suppress result
     bb_rot = RotatedRect(center,  size_initial * scale, rotation/CV_PI * 180);
 
+	// Compute tracking confidence
+	confidence = (float)points_fused.size() / points_initial.size();
+
     //Remember current image
     im_prev = im_gray;
 

diff --git a/CMT.h b/CMT.h
@@ -36,12 +36,14 @@ class CMT
 
     vector<Point2f> points_active; //public for visualization purposes
     RotatedRect bb_rot;
+	float confidence;
 
 private:
     Ptr<FeatureDetector> detector;
     Ptr<DescriptorExtractor> descriptor;
 
     Size2f size_initial;
+	vector<Point2f> points_initial;
 
     vector<int> classes_active;