urban3d_master

building extraction, pytorch, UNet

dataset

The dataset must contain images and masks. You can obtain from there. https://spacenetchallenge.github.io/datasets/Urban_3D_Challenge_summary.html

model

You can choose one of these models, including Unet, Linknet, FPN, PSPNet.

PS

If you want to use this code directly, just do it.Before running this code, you must set the path of your dataset in the specified directory, such as "./data/train_images".