Road Extraction with Satellite Images and Partial Road Maps

Accepted by TGRS'23.

Requirements

• PyTorch 1.7.0

Dataset

• The SpaceNet dataset could be downloaded from this link.
• The OSM dataset could be downloaded from this link.

> cd ..
> mkdir data
> cd data
> mkdir osm
> mkdir spacenet
# Download the datasets into corresponding folders and unzip.

Pretrained Models

• P2CNet trained on mix SpaceNet and OSM datasets could be downloaded from this link.

Test

> python test_deeplabv3plus_mix_mp_sat_gsam_{spacenet,osm}.py

Train

> python train_deeplabv3plus_mix_mp_sat_gsam_{spacenet,osm}.py