CBRNet

A Coarse-to-fine Boundary Refinement Network for Building Extraction from Remote Sensing Imagery

Dataset

WHU Building Dataset
Massachusetts buildings dataset
Inria aerial image dataset

The code

Requirements

• torch
• torchvision
• pillow
• cv2

Usage

Clone the repository:git clone https://github.com/HaonanGuo/CBRNet.git

1. Run s1_offset_generator.py to generate dataset
2. Run s2_Train_CBRNet.py to train CBR-Net
3. Run s3_Eval_CBRNet.py to evaluate the performance of CBR-Net

Help

Any question? Please contact us with: guohnwhu@163.com