├── .gitignore ├── LICENSE ├── README.md ├── configs ├── cityscapes_acdc │ ├── ablations │ │ ├── refign_daformer_row1.yaml │ │ ├── refign_daformer_row2.yaml │ │ ├── refign_daformer_row3.yaml │ │ ├── refign_daformer_row4.yaml │ │ ├── refign_daformer_row5.yaml │ │ └── refign_daformer_row6.yaml │ ├── refign_daformer.yaml │ ├── refign_deeplabv2.yaml │ └── refign_hrda_star.yaml ├── cityscapes_darkzurich │ ├── refign_daformer.yaml │ ├── refign_deeplabv2.yaml │ └── refign_hrda_star.yaml ├── cityscapes_robotcar │ ├── refign_daformer.yaml │ └── refign_deeplabv2.yaml └── megadepth │ ├── uawarpc_evalonly.yaml │ ├── uawarpc_stage1.yaml │ └── uawarpc_stage2.yaml ├── data_modules ├── __init__.py ├── combined_data_module.py ├── datasets │ ├── __init__.py │ ├── acdc.py │ ├── bdd100knight.py │ ├── cityscapes.py │ ├── darkzurich.py │ ├── lists │ │ ├── images_trainval_night_correct_filenames.txt │ │ ├── test_scenes_MegaDepth.txt │ │ ├── train_debug_scenes_MegaDepth.txt │ │ ├── train_scenes_MegaDepth.txt │ │ ├── validation_scenes_MegaDepth.txt │ │ └── zurich_dn_pair_train.csv │ ├── megadepth.py │ ├── nighttimedriving.py │ ├── robotcar.py │ └── robotcarmatching.py └── transforms.py ├── docs └── method.png ├── helpers ├── __init__.py ├── callbacks.py ├── cli.py ├── dacs_transforms.py ├── lr_scheduler.py ├── matching_utils.py ├── metrics.py └── utils.py ├── models ├── __init__.py ├── alignment_model.py ├── backbones │ ├── __init__.py │ ├── mix_transformer.py │ ├── resnet.py │ └── vgg.py ├── correlation_ops │ ├── __init__.py │ ├── correlation.cpp │ ├── correlation_cuda_kernel.cu │ ├── correlation_function.py │ ├── correlation_sampler.cpp │ └── correlation_sampler_cpu.cpp ├── heads │ ├── __init__.py │ ├── base.py │ ├── daformer.py │ ├── deeplabv2.py │ ├── segformer.py │ └── uawarpc.py ├── hrda.py ├── losses.py ├── modules.py └── segmentation_model.py ├── requirements.txt ├── setup.py └── tools ├── convert_cityscapes.py └── run.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/README.md -------------------------------------------------------------------------------- /configs/cityscapes_acdc/ablations/refign_daformer_row1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_acdc/ablations/refign_daformer_row1.yaml -------------------------------------------------------------------------------- /configs/cityscapes_acdc/ablations/refign_daformer_row2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_acdc/ablations/refign_daformer_row2.yaml -------------------------------------------------------------------------------- /configs/cityscapes_acdc/ablations/refign_daformer_row3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_acdc/ablations/refign_daformer_row3.yaml -------------------------------------------------------------------------------- /configs/cityscapes_acdc/ablations/refign_daformer_row4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_acdc/ablations/refign_daformer_row4.yaml -------------------------------------------------------------------------------- /configs/cityscapes_acdc/ablations/refign_daformer_row5.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_acdc/ablations/refign_daformer_row5.yaml -------------------------------------------------------------------------------- /configs/cityscapes_acdc/ablations/refign_daformer_row6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_acdc/ablations/refign_daformer_row6.yaml -------------------------------------------------------------------------------- /configs/cityscapes_acdc/refign_daformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_acdc/refign_daformer.yaml -------------------------------------------------------------------------------- /configs/cityscapes_acdc/refign_deeplabv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_acdc/refign_deeplabv2.yaml -------------------------------------------------------------------------------- /configs/cityscapes_acdc/refign_hrda_star.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_acdc/refign_hrda_star.yaml -------------------------------------------------------------------------------- /configs/cityscapes_darkzurich/refign_daformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_darkzurich/refign_daformer.yaml -------------------------------------------------------------------------------- /configs/cityscapes_darkzurich/refign_deeplabv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_darkzurich/refign_deeplabv2.yaml -------------------------------------------------------------------------------- /configs/cityscapes_darkzurich/refign_hrda_star.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_darkzurich/refign_hrda_star.yaml -------------------------------------------------------------------------------- /configs/cityscapes_robotcar/refign_daformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_robotcar/refign_daformer.yaml -------------------------------------------------------------------------------- /configs/cityscapes_robotcar/refign_deeplabv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/cityscapes_robotcar/refign_deeplabv2.yaml -------------------------------------------------------------------------------- /configs/megadepth/uawarpc_evalonly.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/megadepth/uawarpc_evalonly.yaml -------------------------------------------------------------------------------- /configs/megadepth/uawarpc_stage1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/megadepth/uawarpc_stage1.yaml -------------------------------------------------------------------------------- /configs/megadepth/uawarpc_stage2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/configs/megadepth/uawarpc_stage2.yaml -------------------------------------------------------------------------------- /data_modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/__init__.py -------------------------------------------------------------------------------- /data_modules/combined_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/combined_data_module.py -------------------------------------------------------------------------------- /data_modules/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/__init__.py -------------------------------------------------------------------------------- /data_modules/datasets/acdc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/acdc.py -------------------------------------------------------------------------------- /data_modules/datasets/bdd100knight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/bdd100knight.py -------------------------------------------------------------------------------- /data_modules/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/cityscapes.py -------------------------------------------------------------------------------- /data_modules/datasets/darkzurich.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/darkzurich.py -------------------------------------------------------------------------------- /data_modules/datasets/lists/images_trainval_night_correct_filenames.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/lists/images_trainval_night_correct_filenames.txt -------------------------------------------------------------------------------- /data_modules/datasets/lists/test_scenes_MegaDepth.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/lists/test_scenes_MegaDepth.txt -------------------------------------------------------------------------------- /data_modules/datasets/lists/train_debug_scenes_MegaDepth.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/lists/train_debug_scenes_MegaDepth.txt -------------------------------------------------------------------------------- /data_modules/datasets/lists/train_scenes_MegaDepth.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/lists/train_scenes_MegaDepth.txt -------------------------------------------------------------------------------- /data_modules/datasets/lists/validation_scenes_MegaDepth.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/lists/validation_scenes_MegaDepth.txt -------------------------------------------------------------------------------- /data_modules/datasets/lists/zurich_dn_pair_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/lists/zurich_dn_pair_train.csv -------------------------------------------------------------------------------- /data_modules/datasets/megadepth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/megadepth.py -------------------------------------------------------------------------------- /data_modules/datasets/nighttimedriving.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/nighttimedriving.py -------------------------------------------------------------------------------- /data_modules/datasets/robotcar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/robotcar.py -------------------------------------------------------------------------------- /data_modules/datasets/robotcarmatching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/datasets/robotcarmatching.py -------------------------------------------------------------------------------- /data_modules/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/data_modules/transforms.py -------------------------------------------------------------------------------- /docs/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/docs/method.png -------------------------------------------------------------------------------- /helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /helpers/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/helpers/callbacks.py -------------------------------------------------------------------------------- /helpers/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/helpers/cli.py -------------------------------------------------------------------------------- /helpers/dacs_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/helpers/dacs_transforms.py -------------------------------------------------------------------------------- /helpers/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/helpers/lr_scheduler.py -------------------------------------------------------------------------------- /helpers/matching_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/helpers/matching_utils.py -------------------------------------------------------------------------------- /helpers/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/helpers/metrics.py -------------------------------------------------------------------------------- /helpers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/helpers/utils.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/alignment_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/alignment_model.py -------------------------------------------------------------------------------- /models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/backbones/__init__.py -------------------------------------------------------------------------------- /models/backbones/mix_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/backbones/mix_transformer.py -------------------------------------------------------------------------------- /models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/backbones/resnet.py -------------------------------------------------------------------------------- /models/backbones/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/backbones/vgg.py -------------------------------------------------------------------------------- /models/correlation_ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/correlation_ops/__init__.py -------------------------------------------------------------------------------- /models/correlation_ops/correlation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/correlation_ops/correlation.cpp -------------------------------------------------------------------------------- /models/correlation_ops/correlation_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/correlation_ops/correlation_cuda_kernel.cu -------------------------------------------------------------------------------- /models/correlation_ops/correlation_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/correlation_ops/correlation_function.py -------------------------------------------------------------------------------- /models/correlation_ops/correlation_sampler.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/correlation_ops/correlation_sampler.cpp -------------------------------------------------------------------------------- /models/correlation_ops/correlation_sampler_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/correlation_ops/correlation_sampler_cpu.cpp -------------------------------------------------------------------------------- /models/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/heads/__init__.py -------------------------------------------------------------------------------- /models/heads/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/heads/base.py -------------------------------------------------------------------------------- /models/heads/daformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/heads/daformer.py -------------------------------------------------------------------------------- /models/heads/deeplabv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/heads/deeplabv2.py -------------------------------------------------------------------------------- /models/heads/segformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/heads/segformer.py -------------------------------------------------------------------------------- /models/heads/uawarpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/heads/uawarpc.py -------------------------------------------------------------------------------- /models/hrda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/hrda.py -------------------------------------------------------------------------------- /models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/losses.py -------------------------------------------------------------------------------- /models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/modules.py -------------------------------------------------------------------------------- /models/segmentation_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/models/segmentation_model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/setup.py -------------------------------------------------------------------------------- /tools/convert_cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/tools/convert_cityscapes.py -------------------------------------------------------------------------------- /tools/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/brdav/refign/HEAD/tools/run.py --------------------------------------------------------------------------------