This repository was archived by the owner on Jul 8, 2024. It is now read-only.
- Notifications
You must be signed in to change notification settings - Fork8
MasterSkepticista/Rooftop-Instance-Segmentation
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
Sample Outputs (downsized):
.png)
.png)
.png)
Link:https://www.airs-dataset.com/
Extract the dataset completely, store in this format:
# Training imagesroot/data/train/image# Labelsroot/data/train/label# Test samplesroot/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.
pipinstall-rrequirements.txt
Trained on Nvidia Quadro GP100 with 16GB VRAM. Batch Size:2 Input resolution 3584x3584.
Training Time: 12 hours