Skip to content

Image segmentation of truck parking lots based on coordinates

License

Notifications You must be signed in to change notification settings

f-krause/TruckParkingDetection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

61 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Overview

Starting Image

This repository is based on a project conducted by Michael Fixl, Josef Hinterleitner, Felix Krause and Adrian Seiß at the Vienna University of Economics and Business in collaboration with Trustbit.


The notebooks above perform the following tasks:
  1. Script to manually filter satellite images and labels based on a list of coordinates (of GeoPandas polygon type; coordinates of truck parking lots can be extracted from OpenStreetMap, e.g. via Google BigQuery)
  2. Combine several white- and blacklists from manual filtering to one csv containing coordinates and an unique id
  3. Create training data for object detection (XML file with pixel values of bounding box; object detection not in this repo yet)
  4. Create training data for polygonal labels
    1. Save information in one JSON (necessary for instance segmentation with for example Mask R-CNN)
    2. Save information as masks (necessary for algorithms of Segmentation Models API)
  5. Train image segmentation models (U-Net, Linknet, FPN and PSPNet) in a simple baseline setting
  6. Optimize the best performing model of notebook 4.
  7. Final function, where one can enter coordinates (of a parking lot), and the function returns the parking lot prediction mask for this location
  8. Final function but for public use (only one interactive code chunk)

Segmentation Sample


Data sources and references:

  • Coordinate data from OpenStreetMap
  • Satellite imagery from Esri, World Imagery (Image sources: Esri, Maxar, Earthstar Geographics, CNES/Airbus DS, and the GIS User Community)
  • Code snippets from Dr. Vadim Savenkov
  • Image segmentation training notebooks adapted from Dr. Sreenivas Bhattiprolu
  • Image segmentation models via Segmentation Models API

Please conform to the license applied.

About

Image segmentation of truck parking lots based on coordinates

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published