hujilin1229 / traffic4cast

Submission to the iarai traffic4cast competition

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Traffic4cast–Traffic Map Movie Forecasting

by Team MIE-Lab: Henry Martin, Ye Hong, Dominik Bucher, Christian Rupprecht, René Buffat

Howto

Train a model

  • Copy the competition raw data into data_raw folder such that the top-level city folders are visible.
  • Training configuration can be changed in the config_unetfile (e.g., choose a city).
  • Run the training_unet.py file in the traffic4castfolder
  • The training is logged via tensorboard in the runs folder

Create a submission with the pretrained models

  • Download the pretrained models using this link
  • Store the 3 top-level model folders (e.g., UNet_Berlin) in the runs folder (you can overwrite the existing ones)
  • Run the example_create_3modelsubmission.py script.

About

Submission to the iarai traffic4cast competition

License:Apache License 2.0


Languages

Language:Python 100.0%