Skip to content
This repository has been archived by the owner on Jul 8, 2024. It is now read-only.

MasterSkepticista/Rooftop-Instance-Segmentation

Repository files navigation

Rooftop Instance Segmentation using TensorFlow

Aerial Imagery Dataset provided by National Topographic Office of New Zealand

Sample Outputs (downsized):

Result Result Result

Link: https://www.airs-dataset.com/

Extract the dataset completely, store in this format:

# Training images
root/data/train/image
# Labels
root/data/train/label
# Test samples
root/data/test

Model Used: VGG-16, Instance Segmentation

The script was written on older version of TensorFlow (1.15.x and lower).

Some newer python versions do not include the listing of older TF.

You can downgrade to Python 3.6.x to use it, or use a python virtualenv to install specific python binary (recommended).

Install deps with the following command.

pip install -r requirements.txt

Trained on Nvidia Quadro GP100 with 16GB VRAM. Batch Size:2 Input resolution 3584x3584.

Training Time: 12 hours

Releases

No releases published

Packages

No packages published

Languages