Geoseg - A Computer Vision Package for Automatic Building Segmentation and Outline extraction
Table of Contents
Requirements
- Pytorch == 0.4.1
- Python 3
Organization
Geoseg
โโโ data/
โย ย โโโ original image tiles
โโโ dataset/
โย ย โโโ image&mask slices from data
โโโ checkpoint/
โย ย โโโ pre-trained models
โโโ logs/
โ โโโ curve
โ โโโ raw
โ โโโ snapshot
โย ย speed.csv
โโโ result/
โย ย โโโ quantitative & qualitative result
โโโ src/
โโโ __init__.py
โโโ models
โ โโโ network archs. FCNs, UNet, etc.
โโโ estrain.py
โโโ losses.py
โโโ metrics.py
โโโ runner.py
โโโ test.py
โโโ train.py
โโโ vision.py
Models
Usage
- Download repo.
git clone https://github.com/huster-wgm/geoseg.git
- Download data => NZ32km2
- Download data => Vaihingen
Details about the datasets can be found at Citation.
- Download pre-trainded models (FCNs)
- On NZ32km2(Binary building segmentation) Google Drive
- On ISPRS Vaihingen (6-class segmentation) Google Drive
(Only FCN8s, 16s, and 32s. Others here)
- Step-by-step tutorial
Jupyter-notebook LINK
Performance
Visualization
TODO
- Update training & testing data
- Add support for more dataset
Citation
- NZ32km2 dataset
The location, scale, resolution and preprocessing of the NZ32km2 dataset please refer to paper.LINK
@article{wu2018boundary,
title={A boundary regulated network for accurate roof segmentation and outline extraction},
author={Wu, Guangming and Guo, Zhiling and Shi, Xiaodan and Chen, Qi and Xu, Yongwei and Shibasaki, Ryosuke and Shao, Xiaowei},
journal={Remote Sensing},
volume={10},
number={8},
pages={1195},
year={2018},
publisher={Multidisciplinary Digital Publishing Institute}
}
- ISPRS Vaihingen dataset
The location, scale, resolution and preprocessingof the ISPRS Vaihingen dataset please refer to paper.LINK
@article{wu2019stacked,
title={A Stacked Fully Convolutional Networks with Feature Alignment Framework for Multi-Label Land-cover Segmentation},
author={Wu, Guangming and Guo, Yimin and Song, Xiaoya and Guo, Zhiling and Zhang, Haoran and Shi, Xiaodan and Shibasaki, Ryosuke and Shao, Xiaowei},
journal={Remote Sensing},
volume={11},
number={9},
pages={1051},
year={2019},
publisher={Multidisciplinary Digital Publishing Institute}
}
- Source code
If you use the code for your research, please cite the paper.LINK
@article{wu2018geoseg,
title={Geoseg: A Computer Vision Package for Automatic Building Segmentation and Outline Extraction},
author={Wu, Guangming and Guo, Zhiling},
journal={arXiv preprint arXiv:1809.03175},
year={2018}
}