├── .gitignore ├── README.md ├── code ├── README.md ├── assets │ ├── bg.png │ └── car.png ├── configs │ ├── kitti │ │ ├── kitti_l1.yml │ │ └── kitti_l1_corr.yml │ └── toy │ │ ├── toy_l1.yml │ │ ├── toy_l1_corr.yml │ │ ├── toy_mse.yml │ │ ├── toy_mse_corr.yml │ │ ├── toy_perceptual.yml │ │ └── toy_perceptual_corr.yml ├── corrwise │ ├── corrwise.py │ ├── download_models.sh │ ├── flow_models │ │ ├── README.md │ │ └── raft │ │ │ ├── __init__.py │ │ │ ├── alt_cuda_corr │ │ │ ├── correlation.cpp │ │ │ ├── correlation_kernel.cu │ │ │ └── setup.py │ │ │ ├── corr_raft.py │ │ │ ├── datasets.py │ │ │ ├── extractor.py │ │ │ ├── raft.py │ │ │ ├── raft_utils │ │ │ ├── __init__.py │ │ │ ├── augmentor.py │ │ │ ├── flow_viz.py │ │ │ ├── frame_utils.py │ │ │ └── utils.py │ │ │ └── update.py │ └── flow_utils.py ├── data │ ├── base_data_loader.py │ ├── base_dataset.py │ ├── custom_dataset_data_loader.py │ ├── data_loader.py │ ├── kitti_dataset.py │ └── toy_dataset.py ├── environment.yml ├── eval.py ├── models │ ├── base_model.py │ ├── models.py │ ├── networks.py │ └── simple_model.py ├── options │ ├── base_options.py │ ├── test_options.py │ └── train_options.py ├── requirements.txt ├── scripts │ ├── make_kitti.py │ ├── splits_test.txt │ ├── splits_train.txt │ └── splits_val.txt ├── test.py ├── train.py ├── train_toy.py └── util │ ├── flow_viz.py │ ├── html.py │ ├── util.py │ └── visualizer.py └── docs ├── .gitignore ├── index.html └── resources ├── bibtex.txt ├── paper.png ├── teaser.jpg ├── teaser.mp4 ├── training.mp4 └── viz ├── 1 ├── .DS_Store ├── ctx_1.png ├── ctx_2.png ├── ctx_3.png ├── gen_corr.png └── gen_pix.png ├── 2 ├── ctx_1.png ├── ctx_2.png ├── ctx_3.png ├── gen_corr.png └── gen_pix.png └── toy ├── l1.png ├── l1_corr.png ├── l1p.png ├── mse.png └── sample.png /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/README.md -------------------------------------------------------------------------------- /code/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/README.md -------------------------------------------------------------------------------- /code/assets/bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/assets/bg.png -------------------------------------------------------------------------------- /code/assets/car.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/assets/car.png -------------------------------------------------------------------------------- /code/configs/kitti/kitti_l1.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/configs/kitti/kitti_l1.yml -------------------------------------------------------------------------------- /code/configs/kitti/kitti_l1_corr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/configs/kitti/kitti_l1_corr.yml -------------------------------------------------------------------------------- /code/configs/toy/toy_l1.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/configs/toy/toy_l1.yml -------------------------------------------------------------------------------- /code/configs/toy/toy_l1_corr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/configs/toy/toy_l1_corr.yml -------------------------------------------------------------------------------- /code/configs/toy/toy_mse.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/configs/toy/toy_mse.yml -------------------------------------------------------------------------------- /code/configs/toy/toy_mse_corr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/configs/toy/toy_mse_corr.yml -------------------------------------------------------------------------------- /code/configs/toy/toy_perceptual.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/configs/toy/toy_perceptual.yml -------------------------------------------------------------------------------- /code/configs/toy/toy_perceptual_corr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/configs/toy/toy_perceptual_corr.yml -------------------------------------------------------------------------------- /code/corrwise/corrwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/corrwise.py -------------------------------------------------------------------------------- /code/corrwise/download_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/download_models.sh -------------------------------------------------------------------------------- /code/corrwise/flow_models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/README.md -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/alt_cuda_corr/correlation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/alt_cuda_corr/correlation.cpp -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/alt_cuda_corr/correlation_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/alt_cuda_corr/correlation_kernel.cu -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/alt_cuda_corr/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/alt_cuda_corr/setup.py -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/corr_raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/corr_raft.py -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/datasets.py -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/extractor.py -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/raft.py -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/raft_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/raft_utils/augmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/raft_utils/augmentor.py -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/raft_utils/flow_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/raft_utils/flow_viz.py -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/raft_utils/frame_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/raft_utils/frame_utils.py -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/raft_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/raft_utils/utils.py -------------------------------------------------------------------------------- /code/corrwise/flow_models/raft/update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_models/raft/update.py -------------------------------------------------------------------------------- /code/corrwise/flow_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/corrwise/flow_utils.py -------------------------------------------------------------------------------- /code/data/base_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/data/base_data_loader.py -------------------------------------------------------------------------------- /code/data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/data/base_dataset.py -------------------------------------------------------------------------------- /code/data/custom_dataset_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/data/custom_dataset_data_loader.py -------------------------------------------------------------------------------- /code/data/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/data/data_loader.py -------------------------------------------------------------------------------- /code/data/kitti_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/data/kitti_dataset.py -------------------------------------------------------------------------------- /code/data/toy_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/data/toy_dataset.py -------------------------------------------------------------------------------- /code/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/environment.yml -------------------------------------------------------------------------------- /code/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/eval.py -------------------------------------------------------------------------------- /code/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/models/base_model.py -------------------------------------------------------------------------------- /code/models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/models/models.py -------------------------------------------------------------------------------- /code/models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/models/networks.py -------------------------------------------------------------------------------- /code/models/simple_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/models/simple_model.py -------------------------------------------------------------------------------- /code/options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/options/base_options.py -------------------------------------------------------------------------------- /code/options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/options/test_options.py -------------------------------------------------------------------------------- /code/options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/options/train_options.py -------------------------------------------------------------------------------- /code/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/requirements.txt -------------------------------------------------------------------------------- /code/scripts/make_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/scripts/make_kitti.py -------------------------------------------------------------------------------- /code/scripts/splits_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/scripts/splits_test.txt -------------------------------------------------------------------------------- /code/scripts/splits_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/scripts/splits_train.txt -------------------------------------------------------------------------------- /code/scripts/splits_val.txt: -------------------------------------------------------------------------------- 1 | 2011_09_26_drive_0005_sync 2 | -------------------------------------------------------------------------------- /code/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/test.py -------------------------------------------------------------------------------- /code/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/train.py -------------------------------------------------------------------------------- /code/train_toy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/train_toy.py -------------------------------------------------------------------------------- /code/util/flow_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/util/flow_viz.py -------------------------------------------------------------------------------- /code/util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/util/html.py -------------------------------------------------------------------------------- /code/util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/util/util.py -------------------------------------------------------------------------------- /code/util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/code/util/visualizer.py -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/resources/bibtex.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/bibtex.txt -------------------------------------------------------------------------------- /docs/resources/paper.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/paper.png -------------------------------------------------------------------------------- /docs/resources/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/teaser.jpg -------------------------------------------------------------------------------- /docs/resources/teaser.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/teaser.mp4 -------------------------------------------------------------------------------- /docs/resources/training.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/training.mp4 -------------------------------------------------------------------------------- /docs/resources/viz/1/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/1/.DS_Store -------------------------------------------------------------------------------- /docs/resources/viz/1/ctx_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/1/ctx_1.png -------------------------------------------------------------------------------- /docs/resources/viz/1/ctx_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/1/ctx_2.png -------------------------------------------------------------------------------- /docs/resources/viz/1/ctx_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/1/ctx_3.png -------------------------------------------------------------------------------- /docs/resources/viz/1/gen_corr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/1/gen_corr.png -------------------------------------------------------------------------------- /docs/resources/viz/1/gen_pix.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/1/gen_pix.png -------------------------------------------------------------------------------- /docs/resources/viz/2/ctx_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/2/ctx_1.png -------------------------------------------------------------------------------- /docs/resources/viz/2/ctx_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/2/ctx_2.png -------------------------------------------------------------------------------- /docs/resources/viz/2/ctx_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/2/ctx_3.png -------------------------------------------------------------------------------- /docs/resources/viz/2/gen_corr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/2/gen_corr.png -------------------------------------------------------------------------------- /docs/resources/viz/2/gen_pix.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/2/gen_pix.png -------------------------------------------------------------------------------- /docs/resources/viz/toy/l1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/toy/l1.png -------------------------------------------------------------------------------- /docs/resources/viz/toy/l1_corr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/toy/l1_corr.png -------------------------------------------------------------------------------- /docs/resources/viz/toy/l1p.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/toy/l1p.png -------------------------------------------------------------------------------- /docs/resources/viz/toy/mse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/toy/mse.png -------------------------------------------------------------------------------- /docs/resources/viz/toy/sample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dangeng/CorrWiseLosses/HEAD/docs/resources/viz/toy/sample.png --------------------------------------------------------------------------------