DFC2022 Baseline

A simple baseline for the 2022 IEEE GRSS Data Fusion Contest (DFC2022)

This repository uses TorchGeo, PyTorch Lightning, and Segmentation Models PyTorch to train a U-Net to perform semantic segmentation on the DFC2022 dataset. Masks for the holdout set are then predicted and zipped to be submitted. Note that the the baseline is only trained on the small labeled train set containing imagery from the Nice and Nantes Saint-Nazaire regions.

Install packages

pip install -r requirements.txt

Dataset

The dataset can be downloaded at the DFC2022 IEEE DataPort page and unzipped into a root folder. In our case this is data/.

Train

python train.py --config_file conf/dfc2022.yaml

Predict

python predict.py --log_dir checkpoints/version_0/ --predict_on val --output_directory outputs --device cuda
cd outputs && zip -r submission.zip ./

Submit

Upload submission.zip to the evaluation server here. This baseline results in a mIoU of 0.1278 on the heldout validation set and as of 1/12/22 is 3rd place on the leaderboard.

Checkpoints

Checkpoints can be downloaded from the following link