X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=mtp_example.cc;h=92d406285edd10bc151ba8eea832feda102a6a37;hb=c7d5cdf1982dacc5451f79599041b2e95524d3f7;hp=e615cf113e6d8898c548e041f365013ca3bbb590;hpb=68d9c4f175f774d851a32f6069ad04b31924bfac;p=mtp.git

diff --git a/mtp_example.cc b/mtp_example.cc
index e615cf1..92d4062 100644
--- a/mtp_example.cc
+++ b/mtp_example.cc
@@ -1,7 +1,7 @@
 
 /*
- *  mtp is the ``Multi Tracked Path'', an implementation of the
- *  k-shortest path algorithm for multi-target tracking.
+ *  mtp is the ``Multi Tracked Paths'', an implementation of the
+ *  k-shortest paths algorithm for multi-target tracking.
  *
  *  Copyright (c) 2012 Idiap Research Institute, http://www.idiap.ch/
  *  Written by Francois Fleuret <francois.fleuret@idiap.ch>
@@ -22,22 +22,21 @@
  *
  */
 
-// Multi-Tracked Path
-
 #include <iostream>
 #include <fstream>
 
 using namespace std;
 
-#include "tracker.h"
+#include "mtp_tracker.h"
 
 //////////////////////////////////////////////////////////////////////
 
-scalar_t detection_score(scalar_t a, scalar_t b, scalar_t score_noise, scalar_t flip_noise) {
-  if(drand48() > flip_noise) {
-    return a + score_noise * (2.0 * drand48() - 1.0);
+scalar_t noisy_score(scalar_t true_score, scalar_t erroneous_score,
+                     scalar_t score_noise, scalar_t flip_noise) {
+  if(drand48() < flip_noise) {
+    return erroneous_score + score_noise * (2.0 * drand48() - 1.0);
   } else {
-    return b + score_noise * (2.0 * drand48() - 1.0);
+    return true_score + score_noise * (2.0 * drand48() - 1.0);
   }
 }
 
@@ -46,13 +45,12 @@ int main(int argc, char **argv) {
   int nb_time_steps = 8;
   int motion_amplitude = 1;
 
-  Tracker *tracker = new Tracker();
+  MTPTracker *tracker = new MTPTracker();
 
   tracker->allocate(nb_time_steps, nb_locations);
 
-  // We define the spatial structures by stating what are the possible
-  // motions of targets, and what are the entrances and the
-  // exits.
+  // We define the spatial structure by stating what are the possible
+  // motions of targets, and what are the entrances and the exits.
 
   // Here our example is a 1D space with motions from any location to
   // any location less than motion_amplitude away, entrance at
@@ -66,6 +64,9 @@ int main(int argc, char **argv) {
     tracker->exits[nb_locations - 1] = 1;
   }
 
+  tracker->force_empty_first_frame = 0;
+  tracker->force_empty_last_frame = 0;
+
   // We construct the graph corresponding to this structure
 
   tracker->build_graph();
@@ -81,18 +82,18 @@ int main(int argc, char **argv) {
 
   for(int t = 0; t < nb_time_steps; t++) {
     for(int l = 0; l < nb_locations; l++) {
-      tracker->detection_scores[t][l] = detection_score(-1.0, 1.0, score_noise, flip_noise);
+      tracker->detection_scores[t][l] = noisy_score(-1.0, 1.0, score_noise, flip_noise);
     }
   }
 
-  // Then we two targets with the typical local minimum:
+  // Then we add two targets with a typical tracking local minimum
   //
   // * Target A moves from location 0 to the middle, stays there for a
-  //   while, and comes back, and is strongly detected on the first
+  //   while, and comes back. It is strongly detected on the first
   //   half
   //
   // * Target B moves from location nb_locations-1 to the middle, stay
-  //   there for a while, and comes back, and is strongly detected on
+  //   there for a while, and comes back. It is strongly detected on
   //   the second half
 
   int la, lb; // Target locations
@@ -101,13 +102,13 @@ int main(int argc, char **argv) {
     if(t < nb_time_steps/2) {
       la = t;
       lb = nb_locations - 1 - t;
-      sa = detection_score(10.0, -1.0, score_noise, flip_noise);
-      sb = detection_score( 1.0, -1.0, score_noise, flip_noise);
+      sa = noisy_score(10.0, -1.0, score_noise, flip_noise);
+      sb = noisy_score( 1.0, -1.0, score_noise, flip_noise);
     } else {
       la = nb_time_steps - 1 - t;
       lb = t - nb_time_steps + nb_locations;
-      sa = detection_score( 1.0, -1.0, score_noise, flip_noise);
-      sb = detection_score(10.0, -1.0, score_noise, flip_noise);
+      sa = noisy_score( 1.0, -1.0, score_noise, flip_noise);
+      sb = noisy_score(10.0, -1.0, score_noise, flip_noise);
     }
 
     if(la > nb_locations/2 - 1) la = nb_locations/2 - 1;
@@ -117,12 +118,13 @@ int main(int argc, char **argv) {
     tracker->detection_scores[t][lb] = sb;
   }
 
-  { // Write down the tracker setting
+  { // Write down the tracker setting, so that we can use it as an
+    // example for the mtp command line
     ofstream out_tracker("tracker.dat");
     tracker->write(&out_tracker);
   }
 
-  // Does the tracking per se
+  // Performs the tracking per se
 
   tracker->track();