This repository contains the source code for CS680 S20 Project.
Build based on MIE lab's work (https://github.com/mie-lab/traffic4cast)
Refer to project report for details.
- Modify data loader for traffic4cast 2020 challenge
-
Add new models including Res UNet
-
Add a fusion part for static and dynamic information
-
Create 2 directory
data_new
anddata_new_raw
-
Copy the competition raw data into data_new_raw folder such that the top-level city folders are visible.
-
Without static information
- run
training_unet.py
andtraining_deep_resunet.py
- run
-
Fuse static information
- run
training_unet_fusion.py
andtraining_res_unet_fusion.py
- run
- Loss curve
- Error plot