3rd Place Solution of Traffic4Cast 2021 Core Challenge

This is the code for our solution to the NeurIPS 2021 Traffic4Cast Core Challenge.

Paper

Our solution is described in the "Solving Traffic4Cast Competition with U-Net and Temporal Domain Adaptation" paper.

If you wish to cite this code, please do it as follows:

@misc{konyakhin2021solving,
      title={Solving Traffic4Cast Competition with U-Net and Temporal Domain Adaptation}, 
      author={Vsevolod Konyakhin and Nina Lukashina and Aleksei Shpilman},
      year={2021},
      eprint={2111.03421},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

Competition and Demonstration Track @ NeurIPS 2021

Learnt parameters

The models' learnt parameters are available by the link: https://drive.google.com/file/d/1zD0CecX4P3v5ugxaHO2CQW9oX7_D4BCa/view?usp=sharing
Please download the archive and unzip it into the weights folder of the repository, so its structure looks like the following:

├── ...
├── traffic4cast
├── weights
│   ├── densenet                 
│   │   ├── BERLIN_1008_1430_densenet_unet_mse_best_val_loss_2019=78.4303.pth                     
│   │   ├── CHICAGO_1010_1730_densenet_unet_mse_best_val_loss_2019=41.1579.pth
│   │   └── MELBOURNE_1009_1619_densenet_unet_mse_best_val_loss_2019=25.7395.pth    
│   ├── effnetb5
│   │   ├── BERLIN_1008_1430_efficientnetb5_unet_mse_best_val_loss_2019=80.3510.pth    
│   │   ├── CHICAGO_1012_1035_efficientnetb5_unet_mse_best_val_loss_2019=41.6425.pth
│   │   ├── ISTANBUL_1012_2315_efficientnetb5_unet_mse_best_val_loss_2019=55.7918.pth    
│   │   └── MELBOURNE_1010_0058_efficientnetb5_unet_mse_best_val_loss_2019=26.0132.pth    
│   └── unet
│       ├── BERLIN_0806_1425_vanilla_unet_mse_best_val_loss_2019=0.0000_v5.pth    
│       ├── CHICAGO_0805_0038_vanilla_unet_mse_best_val_loss_2019=42.6634.pth
│       ├── ISTANBUL_0805_2317_vanilla_unet_mse_best_val_loss_2019=0.0000_v4.pth
│       └── MELBOURNE_0804_1942_vanilla_unet_mse_best_val_loss_2019=26.7588.pth
├── ...

Submission reproduction

To generate the submission file, please run the following script:

# $1 - absolute path to the dataset, $2 device to run inference
sh submission.sh {absolute path to dataset} {cpu, cuda}
# Launch example
sh submission.sh /root/data/traffic4cast cuda

The above sctipt generates the submission file submission/submission_all_unets_da_none_mpcpm1_mean_temporal_{date}.zip, which gave us the best MSE of 49.379068541527 on the final leaderboard.

3rd Place Solution of the Traffic4Cast Core Challenge @ NeurIPS 2021

Related tags

Overview

3rd Place Solution of Traffic4Cast 2021 Core Challenge

Paper

Learnt parameters

Submission reproduction

Owner

A library for implementing Decentralized Graph Neural Network algorithms.

Codes for our IJCAI21 paper: Dialogue Discourse-Aware Graph Model and Data Augmentation for Meeting Summarization

salabim - discrete event simulation in Python

Resources related to EMNLP 2021 paper "FAME: Feature-Based Adversarial Meta-Embeddings for Robust Input Representations"

Controlling the MicriSpotAI robot from scratch

Degree-Quant: Quantization-Aware Training for Graph Neural Networks.

Official implementation for the paper: Generating Smooth Pose Sequences for Diverse Human Motion Prediction

Rethinking Space-Time Networks with Improved Memory Coverage for Efficient Video Object Segmentation

基于PaddleClas实现垃圾分类，并转换为inference格式用PaddleHub服务端部署

Just playing with getting CLIP Guided Diffusion running locally, rather than having to use colab.

DyStyle: Dynamic Neural Network for Multi-Attribute-Conditioned Style Editing

Scene-Text-Detection-and-Recognition (Pytorch)

Adaout is a practical and flexible regularization method with high generalization and interpretability

AniGAN: Style-Guided Generative Adversarial Networks for Unsupervised Anime Face Generation

Code, environments, and scripts for the paper: "How Private Is Your RL Policy? An Inverse RL Based Analysis Framework"

Official and maintained implementation of the paper "OSS-Net: Memory Efficient High Resolution Semantic Segmentation of 3D Medical Data" [BMVC 2021].

ConformalLayers: A non-linear sequential neural network with associative layers

TransMorph: Transformer for Medical Image Registration

WSDM‘2022: Knowledge Enhanced Sports Game Summarization

Rust bindings for the C++ api of PyTorch.