WHU Building Dataset
Massachusetts buildings dataset
Inria aerial image dataset
- torch
- torchvision
- pillow
- cv2
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
Any question? Please contact us with: guohnwhu@163.com