├── .gitignore ├── LICENSE ├── README.md ├── config ├── pds-coco │ ├── detone-bihome-lr-5e-3.yaml │ ├── detone-orig-lr-5e-3.yaml │ ├── nguyen-orig-lr-5e-3.yaml │ ├── zeng-bihome-lr-1e-3.yaml │ ├── zeng-orig-lr-1e-3.yaml │ ├── zhang-bihome-lr-1e-2.yaml │ └── zhang-orig-lr-1e-2.yaml └── s-coco │ ├── detone-bihome-lr-5e-3.yaml │ ├── detone-orig-lr-5e-3.yaml │ ├── nguyen-orig-lr-5e-3.yaml │ ├── zeng-bihome-lr-1e-3.yaml │ ├── zeng-orig-lr-1e-3.yaml │ ├── zhang-bihome-lr-1e-2.yaml │ └── zhang-orig-lr-1e-2.yaml ├── data ├── cifar10 │ └── .gitkeep ├── clevr_change │ └── .gitkeep └── coco │ └── .gitkeep ├── eval.py ├── fig └── teaser.png ├── requirements.txt ├── src ├── __init__.py ├── backbones │ ├── ContentAware.py │ ├── HomographyNet.py │ ├── ResNet34.py │ ├── Rethinking.py │ └── utils.py ├── data │ ├── __init__.py │ ├── cifar10 │ │ └── dataset.py │ ├── clevr_change │ │ └── dataset.py │ ├── coco │ │ ├── dataset.py │ │ └── preprocess_offline.py │ ├── flir_adas │ │ └── dataset.py │ ├── transforms.py │ └── utils.py ├── heads │ ├── NoOpHead.py │ ├── PerceptualHead.py │ ├── PhotometricHead.py │ ├── TripletHead.py │ ├── __init__.py │ └── ransac_utils.py └── utils │ ├── checkpoint.py │ ├── dist_util.py │ └── model_zoo.py └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/README.md -------------------------------------------------------------------------------- /config/pds-coco/detone-bihome-lr-5e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/pds-coco/detone-bihome-lr-5e-3.yaml -------------------------------------------------------------------------------- /config/pds-coco/detone-orig-lr-5e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/pds-coco/detone-orig-lr-5e-3.yaml -------------------------------------------------------------------------------- /config/pds-coco/nguyen-orig-lr-5e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/pds-coco/nguyen-orig-lr-5e-3.yaml -------------------------------------------------------------------------------- /config/pds-coco/zeng-bihome-lr-1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/pds-coco/zeng-bihome-lr-1e-3.yaml -------------------------------------------------------------------------------- /config/pds-coco/zeng-orig-lr-1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/pds-coco/zeng-orig-lr-1e-3.yaml -------------------------------------------------------------------------------- /config/pds-coco/zhang-bihome-lr-1e-2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/pds-coco/zhang-bihome-lr-1e-2.yaml -------------------------------------------------------------------------------- /config/pds-coco/zhang-orig-lr-1e-2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/pds-coco/zhang-orig-lr-1e-2.yaml -------------------------------------------------------------------------------- /config/s-coco/detone-bihome-lr-5e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/s-coco/detone-bihome-lr-5e-3.yaml -------------------------------------------------------------------------------- /config/s-coco/detone-orig-lr-5e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/s-coco/detone-orig-lr-5e-3.yaml -------------------------------------------------------------------------------- /config/s-coco/nguyen-orig-lr-5e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/s-coco/nguyen-orig-lr-5e-3.yaml -------------------------------------------------------------------------------- /config/s-coco/zeng-bihome-lr-1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/s-coco/zeng-bihome-lr-1e-3.yaml -------------------------------------------------------------------------------- /config/s-coco/zeng-orig-lr-1e-3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/s-coco/zeng-orig-lr-1e-3.yaml -------------------------------------------------------------------------------- /config/s-coco/zhang-bihome-lr-1e-2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/s-coco/zhang-bihome-lr-1e-2.yaml -------------------------------------------------------------------------------- /config/s-coco/zhang-orig-lr-1e-2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/config/s-coco/zhang-orig-lr-1e-2.yaml -------------------------------------------------------------------------------- /data/cifar10/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/clevr_change/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/coco/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/eval.py -------------------------------------------------------------------------------- /fig/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/fig/teaser.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/backbones/ContentAware.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/backbones/ContentAware.py -------------------------------------------------------------------------------- /src/backbones/HomographyNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/backbones/HomographyNet.py -------------------------------------------------------------------------------- /src/backbones/ResNet34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/backbones/ResNet34.py -------------------------------------------------------------------------------- /src/backbones/Rethinking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/backbones/Rethinking.py -------------------------------------------------------------------------------- /src/backbones/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/backbones/utils.py -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/cifar10/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/data/cifar10/dataset.py -------------------------------------------------------------------------------- /src/data/clevr_change/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/data/clevr_change/dataset.py -------------------------------------------------------------------------------- /src/data/coco/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/data/coco/dataset.py -------------------------------------------------------------------------------- /src/data/coco/preprocess_offline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/data/coco/preprocess_offline.py -------------------------------------------------------------------------------- /src/data/flir_adas/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/data/flir_adas/dataset.py -------------------------------------------------------------------------------- /src/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/data/transforms.py -------------------------------------------------------------------------------- /src/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/data/utils.py -------------------------------------------------------------------------------- /src/heads/NoOpHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/heads/NoOpHead.py -------------------------------------------------------------------------------- /src/heads/PerceptualHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/heads/PerceptualHead.py -------------------------------------------------------------------------------- /src/heads/PhotometricHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/heads/PhotometricHead.py -------------------------------------------------------------------------------- /src/heads/TripletHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/heads/TripletHead.py -------------------------------------------------------------------------------- /src/heads/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/heads/ransac_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/heads/ransac_utils.py -------------------------------------------------------------------------------- /src/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/utils/checkpoint.py -------------------------------------------------------------------------------- /src/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/utils/dist_util.py -------------------------------------------------------------------------------- /src/utils/model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/src/utils/model_zoo.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeurAI-Lab/biHomE/HEAD/train.py --------------------------------------------------------------------------------