tracking_metrics.h

Go to the documentation of this file.

/* =========================================================================== *
   |
   |  Copyright (c) 1994-2011 by Kobus Barnard (author)
   |
   |  Personal and educational use of this code is granted, provided that this
   |  header is kept intact, and that the authorship is not misrepresented, that
   |  its use is acknowledged in publications, and relevant papers are cited.
   |
   |  For other use contact the author (kobus AT cs DOT arizona DOT edu).
   |
   |  Please note that the code in this file has not necessarily been adequately
   |  tested. Naturally, there is no guarantee of performance, support, or fitness
   |  for any particular task. Nonetheless, I am interested in hearing about
   |  problems that you encounter.
   |
   |  Author:  Kyle Simek, Jinyan Guan, Ernesto Brau
 * =========================================================================== */

/* $Id: tracking_metrics.h 18005 2014-10-30 20:45:03Z ksimek $ */

#ifndef TRACKING_METRICS_H
#define TRACKING_METRICS_H

#include <m_cpp/m_matrix.h>
#include <tracking_cpp/tracking_entity.h>
#include <tracking_cpp/tracking_trajectory.h>
#include <boost/bimap.hpp>
#include <vector>

namespace kjb {
namespace tracking {

typedef boost::bimap<Entity_id, Entity_id> Correspondence;

std::vector<Correspondence> get_correspondence 
(
    const Canonical_trajectory_map& gt_trajs,
    const Canonical_trajectory_map& hyp_trajs,
    double threshold = 1.0
);

void get_mt_ml_fragment_and_id_switch
(
    const std::vector<Correspondence>& corrs,
    const Canonical_trajectory_map& gt_trajs,
    const Canonical_trajectory_map& hyp_trajs,
    double& mt,
    double& ml,
    size_t& frags,
    size_t& id_switch
);

void get_mota_and_motp
(
    const std::vector<size_t>& mme_ct,
    const std::vector<size_t>& fp_ct,
    const std::vector<size_t>& miss_ct,
    const std::vector<size_t>& match_ct,
    const std::vector<size_t>& obj_ct,
    const std::vector<double>& dists,
    double& mota,
    double& motp
);

void init_ids_and_points
(
    std::vector<Entity_id>& ids,
    std::vector<const Vector*>& points,
    const Canonical_trajectory_map& trajs,
    size_t cur_frame,
    const Correspondence& existing_corr,
    bool ground_truth
);

void get_pw_distance
(
    const std::vector<const Vector*>& pts1,
    const std::vector<const Vector*>& pts2,
    Matrix& distance
);

void get_best_matching
(
    const kjb::Matrix& pw_distance, 
    double threshold, 
    std::vector<std::pair<int, int> >& matching
);

void get_counts_and_distances
(
    const std::vector<Correspondence>& corrs,
    const Canonical_trajectory_map& gt_trajs,
    const Canonical_trajectory_map& hyp_trajs,
    std::vector<size_t>& mme_ct,
    std::vector<size_t>& fp_ct,
    std::vector<size_t>& miss_ct,
    std::vector<size_t>& match_ct,
    std::vector<size_t>& obj_ct,
    std::vector<double>& dists
);

}} // namespace kjb::tracking


#endif /* TRACKING_METRICS_H */