/Dstl-Satellite-Imagery-Feature-Detection

Place 18 solution for the Dstl feature detection kaggle challenge https://www.kaggle.com/c/dstl-satellite-imagery-feature-detection

Primary LanguagePythonMIT LicenseMIT

Dstl-Satellite-Imagery-Feature-Detection

Place 18 solution for the Dstl feature detection kaggle challenge from DeepVoltaire and Hao

The goal was to find ten potentially overlapping features (buildings, other structures, roads, tracks, trees, crops, rivers, lakes, trucks, cars) in satellite images. This solution uses the U-Net neural network architecture to segment the images for ten binary classes.

Example input image

Raw training image

Example output feature detection

Binary segmentations

To reconstruct the solution

  • Put all data from Kaggle into data/
  • run Preprocessing.py, then training.py and finally submission.py for a good solution from one single model
  • to improve, run the same U-Net model for several classes individually (change the output to only include one class). I used additional single models for buildings, structures, tracks and trees, the other predictions performed better from the 10 class model.

Thanks to visoft, n01z3, Sergey Mushinskiy, Konstantin Lopuhin for the great scripts and discussions.