├── .gitignore ├── LICENSE ├── README.md ├── assets └── logo.png ├── docs ├── benchmark.md ├── demo.md ├── prepare_data.md └── train.md ├── geobench ├── __init__.py ├── modeling │ ├── __init__.py │ ├── archs │ │ ├── __init__.py │ │ ├── dam2 │ │ │ ├── __init__.py │ │ │ ├── dam2.py │ │ │ └── util │ │ │ │ ├── blocks.py │ │ │ │ └── transform.py │ │ ├── dsine │ │ │ ├── __init__.py │ │ │ ├── dsine.py │ │ │ └── submodules.py │ │ └── metric3d │ │ │ ├── __init__.py │ │ │ ├── configs │ │ │ ├── HourglassDecoder │ │ │ │ └── convlarge.0.3_150.py │ │ │ ├── RaftDecoder │ │ │ │ ├── vit.raft5.giant2.py │ │ │ │ ├── vit.raft5.large.py │ │ │ │ └── vit.raft5.small.py │ │ │ ├── __init__.py │ │ │ └── _base_ │ │ │ │ ├── _data_base_.py │ │ │ │ ├── datasets │ │ │ │ └── _data_base_.py │ │ │ │ ├── default_runtime.py │ │ │ │ └── models │ │ │ │ ├── backbones │ │ │ │ ├── convnext_large.py │ │ │ │ ├── dino_vit_giant2_reg.py │ │ │ │ ├── dino_vit_large.py │ │ │ │ ├── dino_vit_large_reg.py │ │ │ │ └── dino_vit_small_reg.py │ │ │ │ └── encoder_decoder │ │ │ │ ├── convnext_large.hourglassdecoder.py │ │ │ │ ├── dino_vit_giant2_reg.dpt_raft.py │ │ │ │ ├── dino_vit_large.dpt_raft.py │ │ │ │ ├── dino_vit_large_reg.dpt_raft.py │ │ │ │ └── dino_vit_small_reg.dpt_raft.py │ │ │ ├── decode_heads │ │ │ ├── HourGlassDecoder.py │ │ │ ├── RAFTDepthNormalDPTDecoder5.py │ │ │ └── __init__.py │ │ │ ├── model_pipelines │ │ │ ├── __base_model__.py │ │ │ ├── __init__.py │ │ │ └── dense_pipeline.py │ │ │ ├── monodepth_model.py │ │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── avg_meter.py │ │ │ ├── comm.py │ │ │ ├── custom_data.py │ │ │ ├── do_test.py │ │ │ ├── inverse_warp.py │ │ │ ├── logger.py │ │ │ ├── mldb.py │ │ │ ├── pcd_filter.py │ │ │ ├── running.py │ │ │ ├── transform.py │ │ │ ├── unproj_pcd.py │ │ │ └── visualization.py │ └── backbones │ │ ├── ConvNeXt.py │ │ ├── __init__.py │ │ └── vit │ │ ├── ViT_DINO.py │ │ ├── ViT_DINO_reg.py │ │ ├── __init__.py │ │ ├── dinov2.py │ │ └── dinov2_layers │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── block.py │ │ ├── drop_path.py │ │ ├── layer_scale.py │ │ ├── mlp.py │ │ ├── patch_embed.py │ │ └── swiglu_ffn.py ├── pipelines │ ├── __init__.py │ ├── flow_pipeline.py │ ├── genpercept_pipeline.py │ ├── geowizard_pipeline.py │ ├── marigold_pipeline.py │ └── wonder3d_pipeline.py └── utils │ ├── batchsize.py │ ├── bilateral_normal_integration_cupy.py │ ├── bilateral_normal_integration_numpy.py │ ├── colormap.py │ ├── count_macs_params.py │ ├── cross.py │ ├── de_normalized.py │ ├── depth2normal.py │ ├── depth_ensemble.py │ ├── dist_utils.py │ ├── image_util.py │ ├── lr_scheduler.py │ ├── mmcv_config.py │ ├── multi_res_noise.py │ ├── normal_ensemble.py │ ├── plane_svd.py │ ├── recon_util.py │ ├── rotation.py │ ├── seed_all.py │ ├── text_embeds_sd21.npy │ ├── text_embeds_sdxl.npy │ ├── text_embeds_t5.npy │ └── transforms.py ├── requirements.txt ├── runners ├── run_dam2.py ├── run_marigold.py └── run_metric3d.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/README.md -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/assets/logo.png -------------------------------------------------------------------------------- /docs/benchmark.md: -------------------------------------------------------------------------------- 1 | # TODO -------------------------------------------------------------------------------- /docs/demo.md: -------------------------------------------------------------------------------- 1 | # TODO -------------------------------------------------------------------------------- /docs/prepare_data.md: -------------------------------------------------------------------------------- 1 | # TODO -------------------------------------------------------------------------------- /docs/train.md: -------------------------------------------------------------------------------- 1 | # TODO -------------------------------------------------------------------------------- /geobench/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /geobench/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /geobench/modeling/archs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /geobench/modeling/archs/dam2/__init__.py: -------------------------------------------------------------------------------- 1 | from .dam2 import DepthAnythingV2 -------------------------------------------------------------------------------- /geobench/modeling/archs/dam2/dam2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/dam2/dam2.py -------------------------------------------------------------------------------- /geobench/modeling/archs/dam2/util/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/dam2/util/blocks.py -------------------------------------------------------------------------------- /geobench/modeling/archs/dam2/util/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/dam2/util/transform.py -------------------------------------------------------------------------------- /geobench/modeling/archs/dsine/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /geobench/modeling/archs/dsine/dsine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/dsine/dsine.py -------------------------------------------------------------------------------- /geobench/modeling/archs/dsine/submodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/dsine/submodules.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/__init__.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/HourglassDecoder/convlarge.0.3_150.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/HourglassDecoder/convlarge.0.3_150.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/RaftDecoder/vit.raft5.giant2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/RaftDecoder/vit.raft5.giant2.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/RaftDecoder/vit.raft5.large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/RaftDecoder/vit.raft5.large.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/RaftDecoder/vit.raft5.small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/RaftDecoder/vit.raft5.small.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/_data_base_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/_data_base_.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/datasets/_data_base_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/datasets/_data_base_.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/models/backbones/convnext_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/models/backbones/convnext_large.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/models/backbones/dino_vit_giant2_reg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/models/backbones/dino_vit_giant2_reg.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/models/backbones/dino_vit_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/models/backbones/dino_vit_large.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/models/backbones/dino_vit_large_reg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/models/backbones/dino_vit_large_reg.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/models/backbones/dino_vit_small_reg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/models/backbones/dino_vit_small_reg.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/models/encoder_decoder/convnext_large.hourglassdecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/models/encoder_decoder/convnext_large.hourglassdecoder.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/models/encoder_decoder/dino_vit_giant2_reg.dpt_raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/models/encoder_decoder/dino_vit_giant2_reg.dpt_raft.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/models/encoder_decoder/dino_vit_large.dpt_raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/models/encoder_decoder/dino_vit_large.dpt_raft.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/models/encoder_decoder/dino_vit_large_reg.dpt_raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/models/encoder_decoder/dino_vit_large_reg.dpt_raft.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/configs/_base_/models/encoder_decoder/dino_vit_small_reg.dpt_raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/configs/_base_/models/encoder_decoder/dino_vit_small_reg.dpt_raft.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/decode_heads/HourGlassDecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/decode_heads/HourGlassDecoder.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/decode_heads/RAFTDepthNormalDPTDecoder5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/decode_heads/RAFTDepthNormalDPTDecoder5.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/decode_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/decode_heads/__init__.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/model_pipelines/__base_model__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/model_pipelines/__base_model__.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/model_pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/model_pipelines/__init__.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/model_pipelines/dense_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/model_pipelines/dense_pipeline.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/monodepth_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/monodepth_model.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/avg_meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/avg_meter.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/comm.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/custom_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/custom_data.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/do_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/do_test.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/inverse_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/inverse_warp.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/logger.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/mldb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/mldb.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/pcd_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/pcd_filter.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/running.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/running.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/transform.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/unproj_pcd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/unproj_pcd.py -------------------------------------------------------------------------------- /geobench/modeling/archs/metric3d/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/archs/metric3d/utils/visualization.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/ConvNeXt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/ConvNeXt.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/__init__.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/ViT_DINO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/ViT_DINO.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/ViT_DINO_reg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/ViT_DINO_reg.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/__init__.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/dinov2.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/dinov2_layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/dinov2_layers/__init__.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/dinov2_layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/dinov2_layers/attention.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/dinov2_layers/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/dinov2_layers/block.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/dinov2_layers/drop_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/dinov2_layers/drop_path.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/dinov2_layers/layer_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/dinov2_layers/layer_scale.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/dinov2_layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/dinov2_layers/mlp.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/dinov2_layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/dinov2_layers/patch_embed.py -------------------------------------------------------------------------------- /geobench/modeling/backbones/vit/dinov2_layers/swiglu_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/modeling/backbones/vit/dinov2_layers/swiglu_ffn.py -------------------------------------------------------------------------------- /geobench/pipelines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /geobench/pipelines/flow_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/pipelines/flow_pipeline.py -------------------------------------------------------------------------------- /geobench/pipelines/genpercept_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/pipelines/genpercept_pipeline.py -------------------------------------------------------------------------------- /geobench/pipelines/geowizard_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/pipelines/geowizard_pipeline.py -------------------------------------------------------------------------------- /geobench/pipelines/marigold_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/pipelines/marigold_pipeline.py -------------------------------------------------------------------------------- /geobench/pipelines/wonder3d_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/pipelines/wonder3d_pipeline.py -------------------------------------------------------------------------------- /geobench/utils/batchsize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/batchsize.py -------------------------------------------------------------------------------- /geobench/utils/bilateral_normal_integration_cupy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/bilateral_normal_integration_cupy.py -------------------------------------------------------------------------------- /geobench/utils/bilateral_normal_integration_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/bilateral_normal_integration_numpy.py -------------------------------------------------------------------------------- /geobench/utils/colormap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/colormap.py -------------------------------------------------------------------------------- /geobench/utils/count_macs_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/count_macs_params.py -------------------------------------------------------------------------------- /geobench/utils/cross.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/cross.py -------------------------------------------------------------------------------- /geobench/utils/de_normalized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/de_normalized.py -------------------------------------------------------------------------------- /geobench/utils/depth2normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/depth2normal.py -------------------------------------------------------------------------------- /geobench/utils/depth_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/depth_ensemble.py -------------------------------------------------------------------------------- /geobench/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/dist_utils.py -------------------------------------------------------------------------------- /geobench/utils/image_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/image_util.py -------------------------------------------------------------------------------- /geobench/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/lr_scheduler.py -------------------------------------------------------------------------------- /geobench/utils/mmcv_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/mmcv_config.py -------------------------------------------------------------------------------- /geobench/utils/multi_res_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/multi_res_noise.py -------------------------------------------------------------------------------- /geobench/utils/normal_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/normal_ensemble.py -------------------------------------------------------------------------------- /geobench/utils/plane_svd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/plane_svd.py -------------------------------------------------------------------------------- /geobench/utils/recon_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/recon_util.py -------------------------------------------------------------------------------- /geobench/utils/rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/rotation.py -------------------------------------------------------------------------------- /geobench/utils/seed_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/seed_all.py -------------------------------------------------------------------------------- /geobench/utils/text_embeds_sd21.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/text_embeds_sd21.npy -------------------------------------------------------------------------------- /geobench/utils/text_embeds_sdxl.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/text_embeds_sdxl.npy -------------------------------------------------------------------------------- /geobench/utils/text_embeds_t5.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/text_embeds_t5.npy -------------------------------------------------------------------------------- /geobench/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/geobench/utils/transforms.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/requirements.txt -------------------------------------------------------------------------------- /runners/run_dam2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/runners/run_dam2.py -------------------------------------------------------------------------------- /runners/run_marigold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/runners/run_marigold.py -------------------------------------------------------------------------------- /runners/run_metric3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/runners/run_metric3d.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aim-uofa/GeoBench/HEAD/setup.py --------------------------------------------------------------------------------