A simple baseline for the 2022 IEEE GRSS Data Fusion Contest (DFC2022)
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/.