tqwhdx19/CBRNet

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

Python

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

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

Help

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