Added the fields force_empty_first_frame and force_empty_last_frame in MTPTracker.

[mtp.git] / mtp_tracker.h

diff --git a/mtp_tracker.h b/mtp_tracker.h
index 08de498..1dcb94c 100644 (file)
--- a/mtp_tracker.h
+++ b/mtp_tracker.h
@@ -33,7 +33,6 @@ using namespace std;
 #include "mtp_graph.h"
 
 class MTPTracker {
-  int _nb_locations, _nb_time_steps;
   scalar_t **_detection_score;
   int **_allowed_motion;
   int *_entrances, *_exits;
@@ -47,10 +46,12 @@ class MTPTracker {
 public:
 
   // The spatial structure
+  int nb_locations, nb_time_steps;
   int **allowed_motion;
   int *entrances, *exits;
+  int force_empty_first_frame, force_empty_last_frame;
 
-  // The detection scores at each node
+  // The detection scores at each location and time
   scalar_t **detection_scores;
 
   MTPTracker();