Skip to content

LZC-68/Six_DOF_tracking_evaluation

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

35 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

6DOF_tracking_evaluation

Code to visualize and evaluate the dataset from "A Framework for Evaluating 6-DOF Object Trackers" [arxiv paper].

The dataset can be downloaded at this website.

Dependencies

To train the network, version 0.1 of pytorch_toolbox is required.

Citation

If you use this dataset in your research, please cite:

@inproceedings{garon2018framework,
	       title={A framework for evaluating 6-dof object trackers},
	       author={Garon, Mathieu and Laurendeau, Denis and Lalonde, Jean-Fran{\c{c}}ois},
	       booktitle={Proceedings of the European Conference on Computer Vision (ECCV)},
	       pages={582--597},
	       year={2018}
}

}

Visualize the dataset

Download the sample dataset here (583 MB).

python visualize_sequence -r /path/to/sample -s interaction_hard -o clock

Evaluating

Evaluating a single sequence : two csv files must be provided, one with ground truth poses and one with the predictions. Note that each row represent 16 values of a 4x4 transform matrix.

python evaluate_sequence.py -g /path/to/ground_truth.csv -p /path/to/predictions.csv ```

Evaluating a batch of sequence : the following folder structure is needed:

  • root
    • modelName
      • object_sequence (ex: dragon_interaction_hard)
        • ground_truth.csv
        • prediction_pose.csv
python evaluate_batch.py -r /path/to/root ```

Tracker

Generate the dataset

Change the parameters in generate_dataset.sh. And run to generate the training and validation dataset.

Train the network

Change the parameters in train_deeptrack.sh. And run to train the network.

License

License for Non-Commercial Use

If this software is redistributed, this license must be included.
The term software includes any source files, documentation, executables,
models, and data.

This software is available for general use by academic or non-profit,
or government-sponsored researchers. This license does not grant the
right to use this software or any derivation of it for commercial activities.
For commercial use, please contact Jean-Francois Lalonde at Université Laval
at [email protected].

This software comes with no warranty or guarantee of any kind. By using this
software, the user accepts full liability.

About

Code visualize and evaluate the dataset from "A Framework for Evaluating 6-DOF Object Trackers".

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 98.5%
  • Shell 1.5%