├── README.md ├── code ├── configs │ ├── hiast_setting.yaml │ ├── sl_1.yaml │ ├── sl_2.yaml │ ├── sl_3.yaml │ └── validate.yaml ├── generate_pseudo_labels.py ├── sseg │ ├── __init__.py │ ├── datasets │ │ ├── __init__.py │ │ ├── augmentations.py │ │ ├── loader │ │ │ ├── __init__.py │ │ │ ├── base_dataset.py │ │ │ ├── cityscapes_dataset.py │ │ │ ├── gtav_dataset.py │ │ │ ├── oxford_dataset.py │ │ │ └── synthia_dataset.py │ │ ├── preprocessor.py │ │ └── utils.py │ └── models │ │ ├── __init__.py │ │ ├── modules │ │ ├── __init__.py │ │ ├── discriminator.py │ │ ├── losses.py │ │ ├── resnet.py │ │ ├── schedulers.py │ │ └── seg_models │ │ │ ├── __init__.py │ │ │ └── deeplab_v2.py │ │ └── segmentors │ │ ├── __init__.py │ │ ├── adversarial_warmup_segmentor.py │ │ ├── self_training_segmentor.py │ │ └── source_only_segmentor.py ├── train.py ├── train.sh ├── utils │ ├── __init__.py │ ├── default_config.py │ ├── metrics.py │ ├── registry │ │ ├── __init__.py │ │ ├── register.py │ │ ├── registries.py │ │ └── registry.py │ ├── result_recorder.py │ └── utils.py ├── validate.py └── workflows │ ├── __init__.py │ ├── pseudo_label_generator.py │ ├── trainer │ ├── __init__.py │ ├── adversarial_warmup_trainer.py │ ├── base_trainer.py │ ├── consistency_self_training_trainer.py │ ├── self_training_trainer.py │ └── source_only_trainer.py │ └── validator.py ├── data ├── SYNTHIA_train.json ├── cityscapes_train.json ├── cityscapes_val.json ├── gtav_train.json ├── oxford_train.json └── oxford_val.json ├── imgs └── framework.png └── requirements.txt /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/README.md -------------------------------------------------------------------------------- /code/configs/hiast_setting.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/configs/hiast_setting.yaml -------------------------------------------------------------------------------- /code/configs/sl_1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/configs/sl_1.yaml -------------------------------------------------------------------------------- /code/configs/sl_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/configs/sl_2.yaml -------------------------------------------------------------------------------- /code/configs/sl_3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/configs/sl_3.yaml -------------------------------------------------------------------------------- /code/configs/validate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/configs/validate.yaml -------------------------------------------------------------------------------- /code/generate_pseudo_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/generate_pseudo_labels.py -------------------------------------------------------------------------------- /code/sseg/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -* 2 | -------------------------------------------------------------------------------- /code/sseg/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -* 2 | -------------------------------------------------------------------------------- /code/sseg/datasets/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/datasets/augmentations.py -------------------------------------------------------------------------------- /code/sseg/datasets/loader/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -* 2 | -------------------------------------------------------------------------------- /code/sseg/datasets/loader/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/datasets/loader/base_dataset.py -------------------------------------------------------------------------------- /code/sseg/datasets/loader/cityscapes_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/datasets/loader/cityscapes_dataset.py -------------------------------------------------------------------------------- /code/sseg/datasets/loader/gtav_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/datasets/loader/gtav_dataset.py -------------------------------------------------------------------------------- /code/sseg/datasets/loader/oxford_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/datasets/loader/oxford_dataset.py -------------------------------------------------------------------------------- /code/sseg/datasets/loader/synthia_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/datasets/loader/synthia_dataset.py -------------------------------------------------------------------------------- /code/sseg/datasets/preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/datasets/preprocessor.py -------------------------------------------------------------------------------- /code/sseg/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/datasets/utils.py -------------------------------------------------------------------------------- /code/sseg/models/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -* 2 | -------------------------------------------------------------------------------- /code/sseg/models/modules/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -* 2 | 3 | -------------------------------------------------------------------------------- /code/sseg/models/modules/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/models/modules/discriminator.py -------------------------------------------------------------------------------- /code/sseg/models/modules/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/models/modules/losses.py -------------------------------------------------------------------------------- /code/sseg/models/modules/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/models/modules/resnet.py -------------------------------------------------------------------------------- /code/sseg/models/modules/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/models/modules/schedulers.py -------------------------------------------------------------------------------- /code/sseg/models/modules/seg_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/models/modules/seg_models/__init__.py -------------------------------------------------------------------------------- /code/sseg/models/modules/seg_models/deeplab_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/models/modules/seg_models/deeplab_v2.py -------------------------------------------------------------------------------- /code/sseg/models/segmentors/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -* 2 | -------------------------------------------------------------------------------- /code/sseg/models/segmentors/adversarial_warmup_segmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/models/segmentors/adversarial_warmup_segmentor.py -------------------------------------------------------------------------------- /code/sseg/models/segmentors/self_training_segmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/models/segmentors/self_training_segmentor.py -------------------------------------------------------------------------------- /code/sseg/models/segmentors/source_only_segmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/sseg/models/segmentors/source_only_segmentor.py -------------------------------------------------------------------------------- /code/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/train.py -------------------------------------------------------------------------------- /code/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/train.sh -------------------------------------------------------------------------------- /code/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -* 2 | -------------------------------------------------------------------------------- /code/utils/default_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/utils/default_config.py -------------------------------------------------------------------------------- /code/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/utils/metrics.py -------------------------------------------------------------------------------- /code/utils/registry/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -* 2 | -------------------------------------------------------------------------------- /code/utils/registry/register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/utils/registry/register.py -------------------------------------------------------------------------------- /code/utils/registry/registries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/utils/registry/registries.py -------------------------------------------------------------------------------- /code/utils/registry/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/utils/registry/registry.py -------------------------------------------------------------------------------- /code/utils/result_recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/utils/result_recorder.py -------------------------------------------------------------------------------- /code/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/utils/utils.py -------------------------------------------------------------------------------- /code/validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/validate.py -------------------------------------------------------------------------------- /code/workflows/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -* 2 | -------------------------------------------------------------------------------- /code/workflows/pseudo_label_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/workflows/pseudo_label_generator.py -------------------------------------------------------------------------------- /code/workflows/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -* 2 | -------------------------------------------------------------------------------- /code/workflows/trainer/adversarial_warmup_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/workflows/trainer/adversarial_warmup_trainer.py -------------------------------------------------------------------------------- /code/workflows/trainer/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/workflows/trainer/base_trainer.py -------------------------------------------------------------------------------- /code/workflows/trainer/consistency_self_training_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/workflows/trainer/consistency_self_training_trainer.py -------------------------------------------------------------------------------- /code/workflows/trainer/self_training_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/workflows/trainer/self_training_trainer.py -------------------------------------------------------------------------------- /code/workflows/trainer/source_only_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/workflows/trainer/source_only_trainer.py -------------------------------------------------------------------------------- /code/workflows/validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/code/workflows/validator.py -------------------------------------------------------------------------------- /data/SYNTHIA_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/data/SYNTHIA_train.json -------------------------------------------------------------------------------- /data/cityscapes_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/data/cityscapes_train.json -------------------------------------------------------------------------------- /data/cityscapes_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/data/cityscapes_val.json -------------------------------------------------------------------------------- /data/gtav_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/data/gtav_train.json -------------------------------------------------------------------------------- /data/oxford_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/data/oxford_train.json -------------------------------------------------------------------------------- /data/oxford_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/data/oxford_val.json -------------------------------------------------------------------------------- /imgs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/imgs/framework.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bupt-ai-cz/HIAST/HEAD/requirements.txt --------------------------------------------------------------------------------