├── .gitattributes ├── LICENSE ├── README.md ├── cmd.sh ├── depth_metrics.py ├── main.py ├── midas ├── base_model.py ├── blocks.py ├── dpt_depth.py ├── midas_net.py ├── midas_net_custom.py ├── transforms.py └── vit.py ├── movie ├── movie_midas3(dpt)_after_finetuning(ours).mp4 └── movie_midas3(dpt)_original_not_equivariant.mp4 ├── nyu_finetune.py ├── nyuv2.py ├── omnidata ├── __init__.py ├── augmentation.py ├── masks.py ├── omnidata_dataset.py ├── refocus_augmentation.py ├── splits.py ├── splits │ ├── metadata_images_split_scene_v1.csv │ ├── test_hypersim_orig.csv │ ├── train_hypersim_orig.csv │ ├── train_val_test_blendedMVS.csv │ ├── train_val_test_debug.csv │ ├── train_val_test_full.csv │ ├── train_val_test_fullplus.csv │ ├── train_val_test_gso.csv │ ├── train_val_test_hypersim.csv │ ├── train_val_test_medium.csv │ ├── train_val_test_replica.csv │ ├── train_val_test_tiny.csv │ └── val_hypersim_orig.csv ├── task_configs.py ├── taskonomy_dataset.py └── transforms.py ├── resnet_depth.py └── unet.py /.gitattributes: -------------------------------------------------------------------------------- 1 | * text eol=lf -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/README.md -------------------------------------------------------------------------------- /cmd.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/cmd.sh -------------------------------------------------------------------------------- /depth_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/depth_metrics.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/main.py -------------------------------------------------------------------------------- /midas/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/midas/base_model.py -------------------------------------------------------------------------------- /midas/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/midas/blocks.py -------------------------------------------------------------------------------- /midas/dpt_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/midas/dpt_depth.py -------------------------------------------------------------------------------- /midas/midas_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/midas/midas_net.py -------------------------------------------------------------------------------- /midas/midas_net_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/midas/midas_net_custom.py -------------------------------------------------------------------------------- /midas/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/midas/transforms.py -------------------------------------------------------------------------------- /midas/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/midas/vit.py -------------------------------------------------------------------------------- /movie/movie_midas3(dpt)_after_finetuning(ours).mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/movie/movie_midas3(dpt)_after_finetuning(ours).mp4 -------------------------------------------------------------------------------- /movie/movie_midas3(dpt)_original_not_equivariant.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/movie/movie_midas3(dpt)_original_not_equivariant.mp4 -------------------------------------------------------------------------------- /nyu_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/nyu_finetune.py -------------------------------------------------------------------------------- /nyuv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/nyuv2.py -------------------------------------------------------------------------------- /omnidata/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /omnidata/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/augmentation.py -------------------------------------------------------------------------------- /omnidata/masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/masks.py -------------------------------------------------------------------------------- /omnidata/omnidata_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/omnidata_dataset.py -------------------------------------------------------------------------------- /omnidata/refocus_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/refocus_augmentation.py -------------------------------------------------------------------------------- /omnidata/splits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits.py -------------------------------------------------------------------------------- /omnidata/splits/metadata_images_split_scene_v1.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/metadata_images_split_scene_v1.csv -------------------------------------------------------------------------------- /omnidata/splits/test_hypersim_orig.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/test_hypersim_orig.csv -------------------------------------------------------------------------------- /omnidata/splits/train_hypersim_orig.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/train_hypersim_orig.csv -------------------------------------------------------------------------------- /omnidata/splits/train_val_test_blendedMVS.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/train_val_test_blendedMVS.csv -------------------------------------------------------------------------------- /omnidata/splits/train_val_test_debug.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/train_val_test_debug.csv -------------------------------------------------------------------------------- /omnidata/splits/train_val_test_full.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/train_val_test_full.csv -------------------------------------------------------------------------------- /omnidata/splits/train_val_test_fullplus.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/train_val_test_fullplus.csv -------------------------------------------------------------------------------- /omnidata/splits/train_val_test_gso.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/train_val_test_gso.csv -------------------------------------------------------------------------------- /omnidata/splits/train_val_test_hypersim.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/train_val_test_hypersim.csv -------------------------------------------------------------------------------- /omnidata/splits/train_val_test_medium.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/train_val_test_medium.csv -------------------------------------------------------------------------------- /omnidata/splits/train_val_test_replica.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/train_val_test_replica.csv -------------------------------------------------------------------------------- /omnidata/splits/train_val_test_tiny.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/train_val_test_tiny.csv -------------------------------------------------------------------------------- /omnidata/splits/val_hypersim_orig.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/splits/val_hypersim_orig.csv -------------------------------------------------------------------------------- /omnidata/task_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/task_configs.py -------------------------------------------------------------------------------- /omnidata/taskonomy_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/taskonomy_dataset.py -------------------------------------------------------------------------------- /omnidata/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/omnidata/transforms.py -------------------------------------------------------------------------------- /resnet_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/resnet_depth.py -------------------------------------------------------------------------------- /unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mikuhatsune/equivariance/HEAD/unet.py --------------------------------------------------------------------------------