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.
pip install -r requirements.txt
The dataset can be downloaded at the DFC2022 IEEE DataPort page and unzipped into a root folder. In our case this is
python train.py --config_file conf/dfc2022.yaml
python predict.py --log_dir checkpoints/version_0/ --predict_on val --output_directory outputs --device cuda
cd outputs && zip -r submission.zip ./
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 can be downloaded from the following link