"Unsupervised Domain Adaptation for the Semantic Segmentation of Remote Sensing Images via One-Shot Image-to-Image Translation"
September 3, 2024 · View on GitHub
This repository contains the official implementation of the
"Unsupervised Domain Adaptation for the Semantic Segmentation of Remote Sensing Images via One-Shot Image-to-Image Translation"
in https://ieeexplore.ieee.org/document/10138565.
Requirements:
- Pytorch 1.11.0
* python 3.8.10
+ Semantic segmentation model (Smp V0.2.1)
Dataset:
SPRS dataset: Potsdam IRRG and Vaihingen IRRG with size (512×512)
train:
To train the I2I translation model (step 1), use the I2I_train folder.
To train the Semantic segmentation model (step 2), use the SemSeg_train folder.
The P_IRRG target one-shot images and V_IRRG target one shot images contain the one-shot image that has been used during the train g of the I2IT
test and evaluate:
Use the last three line of the SemSeg_train.py