Code for BrnoCompSpeed dataset
Clone or download
Latest commit 05cde7a May 9, 2018
Type Name Latest commit message Commit time
Failed to load latest commit information.
code different plot styles Jul 31, 2017
dataset checksums Oct 15, 2016
results dataset and results dirs Oct 14, 2016 Update May 9, 2018

BrnoCompSpeed dataset evaluation code

Dataset published with paper SOCHOR Jakub et al. Comprehensive Data Set for Automatic Single Camera Visual Speed Measurement, IEEE T-ITS

The dataset can be downloaded from our website. BibTeX citation is also on the site.

How to use the code

  1. Install PYTHON2, packages you might NOT have: numpy, scipy, matplotlib, tabulate
  2. Download the dataset and place the results and dataset folders from the downloaded archive on the same level as the code folder (root of the repository).
  • TIP: to save disk space use following command to get and unpack the dataset (WARNING: it has ~200GB)
  • curl | tar xv
  1. (Optional) Modify paths in file code/
  2. Check file code/ The most important variables are RUN_FOR_SYSTEMS and RUN_FOR_VIDEOS.
  3. Run in code direcotory: python and wait. The results will be computed, shown and cached in results directory. The script has several arguments, so you can use --help for explanation of the arguments. I STRONGLY recommend to use ipython, spyder or similar terminals.

Additional information

  • For information about vehicle types from our country (Czech Republic), you can use following datasets: COD20K, BoxCars21k.
  • The dataset itself containes videos, mask, screenshots, and pkl file with the ground truth data
  • The dataset contains extra session0 which was annotated manually and is not included in the paper
  • The session0 is meant to be as training for all the splits (A,B,C - see the paper)

How to generate your own result JSON files

  • Just place them into appropriate subdirectory in the results directory

  • The JSON files should have same structure as the already existing JSON files

  • The structure of the files should be following:

      			{"vp1": [x,y], 
      			 "vp2": [x,y], 
                   "pp": [x,y], 
                   "scale": lambda}, (see the paper and supplementary pdf for information how this is used)
              		"id": number,
              		"frames", "posX", "posY": each key defining list of positions of the car in the video
              		posX and posY should contain x,y coordinates of a point of the vehicle which is on the road plane
              		and frames should contain frame numbers of the reported points. The length of the vectors must be equal.
              		Examples can be found in the results directory.