Awesome
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
Download
To download the dataset, reach out to Jakub Špaňhel - ispanhel@fit.vutbr.cz.
How to use the code
- Install PYTHON2, packages you might NOT have:
numpy, scipy, matplotlib, tabulate
- 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 https://medusa.fit.vutbr.cz/traffic/data/2016-ITS-BrnoCompSpeed-full.tar | tar xv
- (Optional) Modify paths in file code/dataset_info.py
- Check file code/config.py. The most important variables are
RUN_FOR_SYSTEMS
andRUN_FOR_VIDEOS
. - Run in code direcotory:
python eval.py
and wait. The results will be computed, shown and cached in results directory. The scripteval.py
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:
{ "camera_calibration": {"vp1": [x,y], "vp2": [x,y], "pp": [x,y], "scale": lambda}, (see the paper and supplementary pdf for information how this is used) "cars":[ { "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. } ] } }
Contact
{isochor,herout,ijuranek}@fit.vutbr.cz