├── .gitignore ├── README.md ├── dataset.py ├── demo └── Spherical_transform.ipynb ├── demo_gradio.py ├── demo_gradio_kitti.py ├── evaluate.py ├── figure ├── Demo.png ├── Example_kitti.png ├── Example_random.png ├── VIGOR_exp │ ├── pano.png │ └── sat.png ├── VIGOR_label.png ├── exp1.jpg ├── exp2.png ├── exp3.jepg └── pipeline.png ├── models ├── config │ └── VIGOR │ │ └── train-vigor.json ├── corr.py ├── efficientnet_pytorch │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── model.cpython-38.pyc │ │ └── utils.cpython-38.pyc │ ├── model.py │ └── utils.py ├── network.py ├── update.py └── utils │ ├── Mercator.py │ ├── __pycache__ │ ├── torch_geometry.cpython-38.pyc │ └── utils.cpython-38.pyc │ ├── augment.py │ ├── loss_factory.py │ ├── torch_geometry.py │ └── utils.py ├── myevaluate.py ├── requirements.txt ├── train.py ├── train.sh └── val.sh /.gitignore: -------------------------------------------------------------------------------- 1 | **/__pycache__ 2 | 3 | watch/ 4 | checkpoints/* 5 | 6 | **/a_local_test -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/README.md -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/dataset.py -------------------------------------------------------------------------------- /demo/Spherical_transform.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/demo/Spherical_transform.ipynb -------------------------------------------------------------------------------- /demo_gradio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/demo_gradio.py -------------------------------------------------------------------------------- /demo_gradio_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/demo_gradio_kitti.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/evaluate.py -------------------------------------------------------------------------------- /figure/Demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/figure/Demo.png -------------------------------------------------------------------------------- /figure/Example_kitti.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/figure/Example_kitti.png -------------------------------------------------------------------------------- /figure/Example_random.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/figure/Example_random.png -------------------------------------------------------------------------------- /figure/VIGOR_exp/pano.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/figure/VIGOR_exp/pano.png -------------------------------------------------------------------------------- /figure/VIGOR_exp/sat.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/figure/VIGOR_exp/sat.png -------------------------------------------------------------------------------- /figure/VIGOR_label.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/figure/VIGOR_label.png -------------------------------------------------------------------------------- /figure/exp1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/figure/exp1.jpg -------------------------------------------------------------------------------- /figure/exp2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/figure/exp2.png -------------------------------------------------------------------------------- /figure/exp3.jepg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/figure/exp3.jepg -------------------------------------------------------------------------------- /figure/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/figure/pipeline.png -------------------------------------------------------------------------------- /models/config/VIGOR/train-vigor.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/config/VIGOR/train-vigor.json -------------------------------------------------------------------------------- /models/corr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/corr.py -------------------------------------------------------------------------------- /models/efficientnet_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/efficientnet_pytorch/__init__.py -------------------------------------------------------------------------------- /models/efficientnet_pytorch/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/efficientnet_pytorch/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /models/efficientnet_pytorch/__pycache__/model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/efficientnet_pytorch/__pycache__/model.cpython-38.pyc -------------------------------------------------------------------------------- /models/efficientnet_pytorch/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/efficientnet_pytorch/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /models/efficientnet_pytorch/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/efficientnet_pytorch/model.py -------------------------------------------------------------------------------- /models/efficientnet_pytorch/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/efficientnet_pytorch/utils.py -------------------------------------------------------------------------------- /models/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/network.py -------------------------------------------------------------------------------- /models/update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/update.py -------------------------------------------------------------------------------- /models/utils/Mercator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/utils/Mercator.py -------------------------------------------------------------------------------- /models/utils/__pycache__/torch_geometry.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/utils/__pycache__/torch_geometry.cpython-38.pyc -------------------------------------------------------------------------------- /models/utils/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/utils/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /models/utils/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/utils/augment.py -------------------------------------------------------------------------------- /models/utils/loss_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/utils/loss_factory.py -------------------------------------------------------------------------------- /models/utils/torch_geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/utils/torch_geometry.py -------------------------------------------------------------------------------- /models/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/models/utils/utils.py -------------------------------------------------------------------------------- /myevaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/myevaluate.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/train.sh -------------------------------------------------------------------------------- /val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xlwangDev/HC-Net/HEAD/val.sh --------------------------------------------------------------------------------