Skip to content

frankfengdi/argoverse-kitti-adapter

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

49 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Python 3.6 Python 3.7

argoverse-kitti-adapter

A toolbox to translate Argoverse dataset (CVPR2019) into KITTI dataset (CVPR2012) format for perception/tracking tasks.

Introduction

This toolbox is to translate Argoverse dataset from CVPR2019 into the KITTI dataset format. The major changes are:

  1. Changing labels from the .json files into the .txt file format (15 columns) used in KITTI dataset;
  2. Reconfigure the calibration files from the .json files per log into .txt file format per image/lidar scan;
  3. Copying related image files and lidar files to form the corrsponding data structure used in KITTI perception tasks;

Usage

1. Download Data

  • Please download the most recent argoverse dataset. Once the data is downloaded, it is convenient to put all the training logs under one training folder. Please do not change any contents under the individual log folder. Here is a document tree as an illustration:
argodataset
└── argoverse-tracking <----------------------------root_dir
    └── train <-------------------------------------data_dir
        └── 0ef28d5c-ae34-370b-99e7-6709e1c4b929
        └── 00c561b9-2057-358d-82c6-5b06d76cebcf
        └── ...
    └── validation
        └──5c251c22-11b2-3278-835c-0cf3cdee3f44
        └──...
    └── test
        └──8a15674a-ae5c-38e2-bc4b-f4156d384072
        └──...

2. Download Argoverse Repo

3. Clone the Argoverse-kitti-adapter Repo

git clone https://github.com/yzhou377/argoverse-kitti-adapter.git

  • Once the data and the API are well equipped, please open the 'apater.py' file for changing your root_dir (the directory to your argoverse-tracking folder). The toolbox will automatically construct a new folder (train_kitti) for you. The new file structure is shown as below:
argodataset
└── argoverse-tracking <----------------------------root_dir
    └── train <-------------------------------------data_dir
        └──5c251c22-11b2-3278-835c-0cf3cdee3f44
        └──...
    └── argoverse-kitti <---------------------------goal_dir
        └── training
           └──velodyne
           └──iamge_2
           └──calib
           └──label_2
        └── statistics <----------------------------dataset statistics
        └── Imagesets <-----------------------------dataset split

4. Change Hyperparameters

  • On the top of the adapter.py file, please change the root directory and the distance threshold.

5. Run the Adapter

  • After changing the configruation file, please run the adapter.py file using the following commands python adapter.py

Note

  1. Frequency and Sychronization
  • In KITTI, the camera and the LIDAR are synchronized at 10Hz. However, in Argoverse, the ring cameras are running at 30Hz, while the LIDAR is running at 10Hz. To fully realize the KITTI dataset format, we match each LIDAR scan with the corresponding camera at the closest timestamp. As a result, the sensor combo in the modified KITTI version of Argoverse is still running at 10Hz.
  1. Labelling File Clips
  • KITTI only labels the object in the view of the front camera, while Argoverse, given its panoramic nature, labels all the obstacles around the object. Thus, for each associated labelling file, if the object is not seen in this specific image, then it is not labelled. Furthermore, objects that are too small (beyond 70m) were not labelled. One can cetrainly change this threshold in the apater.py. Here attaches the KITTI label README file . For the Argoverse label file, please go check the Argoverse github
  1. Calibration File
  • To match the KITTI calibration file, the tool is designed to combine the 'R0_rect' matrix together with the 'P2' matrix to form intrinsic matrix 'K' of the camera. In the new label file, 'R0_rect' is set to be an identity matrix, while 'P2' contains all the intrinsics.

Reference

  • [1] M. Chang et al., Argoverse: 3D Tracking and Forecasting with Rich Maps, CVPR2019, Long Beach, U.S.A
  • [2] A. Geiger et al., Are we ready for Autonomous Driving? The KITTI Vision Benchmark Suite, CVPR2012, Rhode Island, U.S.A
  • [3] Y. Wang et al., Train in Germany, Test in The USA: Making 3D Object Detectors Generalize, CVPR2020, Seatle, U.S.A

About

A tool to translate Argoverse into KITTI dataset format

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%