Skip to content

Vehicle Re-Identification (ReID) dataset contains over 55,000 images for training and validation of the vehicle re-identification model

License

Notifications You must be signed in to change notification settings

sekilab/VehicleReIdentificationDataset

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Vehicle Re-Identification (ReID) Dataset

Sample images from the CARLA ReID dataset, resized to the same dimension for visualization. Figure on the left shows the variation of vehicle models and colors in the CARLA ReID dataset. CARLA ReID dataset contains four classes of vehicles such as car, truck, motorcycle, and bicycle. Figure on the right shows the camera angle variation for observing the same vehicle. Each vehicle is observed by 85 cameras located at varying distances surrounding the target vehicle.

  • The file names contain the information on vehicle id and camera id separated by an underscore, necessary to train ReID model.

    • If the filename is 20220711212617_24_9.jpg, then 20220711212617 is the datetime, 24 is the camera id and 9 represents the vehicle id. These two information is enough for training a ReID model

Download links

The download URL of the VeRi dataset is shown here. Please note that training, gallery and query images are all inside the same zip file.

Download VeRi dataset (CARLA)

Training example and Installation

  • Please check training_vehicle_reid.ipynb Jupyter notebook to train the model or understand the training process, in general.

Installation

  • Make sure that you have conda installed. Let's create a new virtual environment carla_reid with Python verison 3.7 by executing the following command:
conda create -n carla_reid python=3.7 pip pandas  -y
  • The above command, however, will not install torchreid. Also, pip package does not support latest build of torchreid and it needs to built from the source. I have provided bash script to simplify this task

  • First of all, activate the environment created before by executing the following command:

conda activate carla_reid
  • Make sure that the environment is activated and you are in the root directory of this repository. It should show (carla_reid) on the left of your username in the terminal. Run the following command:
bash install_packages_reid.sh
  • The above command will install torchreid 1.4.0 along with dependencies such as torch, torchvision, etc. You are all set to run the Jupyter notebook.

Citations

If you find our dataset useful and use it in your research, please give us a star ⭐ and cite the following article:

@INPROCEEDINGS{10020814,
  author={Kumar, Ashutosh and Kashiyama, Takehiro and Maeda, Hiroya and Zhang, Fan and Omata, Hiroshi and Sekimoto, Yoshihide},
  booktitle={2022 IEEE International Conference on Big Data (Big Data)}, 
  title={Vehicle re-identification and trajectory reconstruction using multiple moving cameras in the CARLA driving simulator}, 
  year={2022},
  volume={},
  number={},
  pages={1858-1865},
  doi={10.1109/BigData55660.2022.10020814}}
@article{kumar2022vehicle,
  title={Vehicle re-identification and trajectory reconstruction using multiple moving cameras in the CARLA driving simulator},
  author={Kumar, Ashutosh and Kashiyama, Takehiro and Maeda, Hiroya and Zhang, Fan and Omata, Hiroshi and Sekimoto, Yoshihide},
  year={2022},
  publisher={TechRxiv}
}