Streetview Segmentation

Perform semantic segmentation on images, using Facebook's Mask2Former models.

1
contributor
29 commitsLast commit ≈ 23 months ago5 stars0 forks

What Streetview Segmentation can do for you

Streetview Segmentation script performs semantic segmentation on images, using models from Facebook's collection of Mask2Former models. Output consists of a CSV-file detailing, for each input image, the number of pixels in each semantic class in that image. Optionally, it can save a copy of each input image overlaid with the semantic segmentation. If the input images are 360° photo's, the script provides the possibility of tranforming them by projecting them onto a cube, resulting in six images per input image.

This script is specifically designed to run on a computer without a GPU. Some of the underlying libraries require the presence of CUDA-drivers to run, even if the actual device is absent. As it can be problematic to install such drivers on a computer without an actual GPU, the program is packaged as Docker-container, based on an official NVIDIA-image, which comes with pre-installed drivers. Building and running the container requires the presence of Docker engine. Note that the container is based on a Linux image (Ubuntu 18.04); running it on a Windows computer may require extra configuration.

Keywords
Programming languages
  • Python 95%
  • Dockerfile 5%
License
Not specified
</>Source code

Participating organisations

Utrecht University

Contributors

MS
Maarten Schermer