LandSlide4Sense 2022 Baseline

Simple U-Net baseline for the LandSlide4Sense 2022 landslide detection challenge. This baseline uses the PyTorch Lightning, Kornia, and Segmentation Models PyTorch packages.

Install Dependencies

pip install -r requirements.txt

Train

python train.py --cfg conf/train.yaml

Predict and Submit

python predict.py --predict_on val \
    --output_directory outputs \
    --device cuda \
    --log_dir logs/version_0
zip submission.zip outputs/*.h5