zoltanszalontay / Solar-Panels-Detection

Detection and location of Solar Panels in aerial images using U-Net

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Solar-Panels-Detection

Scripts

  • src/data_loader.py: classes to load 256x256 images in the training set
  • src/utils/solar_panels_detection_california.py: creation of training set using geojson file and aerial images from here.
  • src/train_unet2.py: training of U-Net using Cuda Tensors
  • src/train_unet2_cpu.py: training of U-Net using cpu Tensors
  • src/Hogwild/train_unet2_cpu_Hogwild.py: distributed training of U-Net in one node of a cluster, doing asynchrnous update of model parameters.
  • src/mpi/train_unet2_cpu_mpi.py: distributed training of U-Net in several nodes of a cluster using mpi4py
  • src/OpenStreetMaps/osm.py: rasterisaton of OpenStreetMaps data to create mask images of Sentinel-2 images. Useful to create training sets for other detection problems

Data

Results

True Positives examples

  • Left: Original image. Center: U-Net output. Right: Solar panels delimited using U-Net output TP1 TP2 TP3 TP4

False Positives Examples

FP1 FP2

Sentinel-2 land monitoring training set creation

  • Sentinel-2 image of Edinburgh area Sentinel2
  • Mask image with white pixels at forests location forest
  • Mask image with white pixels at roads location roads

About

Detection and location of Solar Panels in aerial images using U-Net


Languages

Language:Python 99.7%Language:Shell 0.3%