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

Official PyTorch implementation of "RMGN: A Regional Mask Guided Network for Parser-free Virtual Try-on" (IJCAI-ECAI 2022)

My Body is a Cage: the Role of Morphology in Graph-Based Incompatible Control

A project studying the influence of communication in multi-objective normal-form games

Deep learning for spiking neural networks

This project deals with the detection of skin lesions within the ISICs dataset using YOLOv3 Object Detection with Darknet.

Code for the SIGIR 2022 paper "Hybrid Transformer with Multi-level Fusion for Multimodal Knowledge Graph Completion"

An open-source Deep Learning Engine for Healthcare that aims to treat & prevent major diseases

CRF-RNN for Semantic Image Segmentation - PyTorch version

Code repo for "Cross-Scale Internal Graph Neural Network for Image Super-Resolution" (NeurIPS'20)

Prototype for Baby Action Detection and Classification

This is the face keypoint train code of project face-detection-project

Python scripts performing class agnostic object localization using the Object Localization Network model in ONNX.

Official repo for QHack—the quantum machine learning hackathon

Cross-Task Consistency Learning Framework for Multi-Task Learning

PyTorch implementation of Barlow Twins.

ATAC: Adversarially Trained Actor Critic

CS550 Machine Learning course project on CNN Detection.

A paper using optimal transport to solve the graph matching problem.

Trajectory Extraction of road users via Traffic Camera

Beyond imagenet attack (accepted by ICLR 2022) towards crafting adversarial examples for black-box domains.