Skip to content

The codes and data of paper "Curb-GAN: Conditional Urban Traffic Estimation through Spatio-Temporal Generative Adversarial Networks"

License

Notifications You must be signed in to change notification settings

Curb-GAN/Curb-GAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Curb-GAN

The codes and data of paper "Curb-GAN: Conditional Urban Traffic Estimation through Spatio-Temporal Generative Adversarial Networks".

Overview of Curb_GAN This is the overview of Curb-GAN. The implementation is realized by Pytorch.

Data Preparation

Since the whole dataset is huge, here we only provide sample datasets.

The sample data is located in data folder, please unzip data/speed_data.zip and data/inflow_data.zip and put the corresponding data files into traffic_speed_estimation/ and taxi_inflow_estimation/ folders, respectively.

Requirements for Reproducibility

  • Cuda 9.2
  • pytorch 0.4.1
  • Python 3.6.7
  • Devices: NVIDIA GTX 1080 GPUs
  • System: Ubuntu 16.04

Training

Just take traffic speed estimation as an example:

  1. cd traffic_speed_estimation/.
  2. Running the codes: python train.py. (The parameters are defined in train.py)
  3. The parameters of the trained model will be saved in to traffic_speed_estimation//CurbGAN_train.

The training process for taxi inflow estimation is similar.

About

The codes and data of paper "Curb-GAN: Conditional Urban Traffic Estimation through Spatio-Temporal Generative Adversarial Networks"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages