/satellite-image-segmentation

Image segmentation (roads and houses) on satellite images

Primary LanguageJupyter Notebook

satellite-image-segmentation

Image segmentation (pixel-level binary classification of roads, and separately for houses) on satellite images (miniproject for Harvard AM216 Spring 2022). Application of transfer learning (pretrained Unet encoder), data augmentation, and dealing with class imbalance via DICE loss.

House mask prediction

(True mask on top, predicted mask below) house_preds

Road mask prediction

(True mask on top, predicted mask below) house_preds