Skip to content

isaaccorley/landslide4sense

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Releases

No releases published

Packages

No packages published