ColamanAI commited on
Commit
8b907a2
·
verified ·
1 Parent(s): f41698d

Upload 153 files

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. configs/calibration_benchmark.yaml +23 -0
  2. configs/dataset/ase_wai/default.yaml +3 -0
  3. configs/dataset/ase_wai/train/default.yaml +26 -0
  4. configs/dataset/ase_wai/val/default.yaml +26 -0
  5. configs/dataset/benchmark_512_eth3d_snpp_tav2.yaml +20 -0
  6. configs/dataset/benchmark_518_eth3d_snpp_tav2.yaml +20 -0
  7. configs/dataset/benchmark_sv_calib_518_many_ar_eth3d_snpp_tav2.yaml +20 -0
  8. configs/dataset/blendedmvs_wai/default.yaml +3 -0
  9. configs/dataset/blendedmvs_wai/train/default.yaml +26 -0
  10. configs/dataset/blendedmvs_wai/val/default.yaml +26 -0
  11. configs/dataset/bmvs_518_many_ar_48ipg_8g.yaml +23 -0
  12. configs/dataset/default.yaml +39 -0
  13. configs/dataset/dl3dv_wai/default.yaml +3 -0
  14. configs/dataset/dl3dv_wai/train/default.yaml +28 -0
  15. configs/dataset/dl3dv_wai/val/default.yaml +28 -0
  16. configs/dataset/dynamicreplica_wai/default.yaml +3 -0
  17. configs/dataset/dynamicreplica_wai/train/default.yaml +26 -0
  18. configs/dataset/dynamicreplica_wai/val/default.yaml +26 -0
  19. configs/dataset/eth3d_wai/default.yaml +2 -0
  20. configs/dataset/eth3d_wai/test/default.yaml +22 -0
  21. configs/dataset/megadepth_wai/default.yaml +3 -0
  22. configs/dataset/megadepth_wai/train/default.yaml +26 -0
  23. configs/dataset/megadepth_wai/val/default.yaml +26 -0
  24. configs/dataset/megatrain_13d_512_many_ar_24ipg_16g.yaml +59 -0
  25. configs/dataset/megatrain_13d_518_many_ar_24ipg_16g.yaml +59 -0
  26. configs/dataset/megatrain_13d_518_many_ar_24ipg_8g.yaml +59 -0
  27. configs/dataset/megatrain_13d_518_many_ar_48ipg_64g.yaml +59 -0
  28. configs/dataset/megatrain_13d_518_many_ar_48ipg_8g_mono.yaml +59 -0
  29. configs/dataset/megatrain_6d_518_many_ar_48ipg_64g.yaml +38 -0
  30. configs/dataset/megatrain_6d_518_many_ar_48ipg_8g.yaml +38 -0
  31. configs/dataset/mpsd_wai/default.yaml +3 -0
  32. configs/dataset/mpsd_wai/train/default.yaml +26 -0
  33. configs/dataset/mpsd_wai/val/default.yaml +26 -0
  34. configs/dataset/mvs_synth_wai/default.yaml +3 -0
  35. configs/dataset/mvs_synth_wai/train/default.yaml +26 -0
  36. configs/dataset/mvs_synth_wai/val/default.yaml +26 -0
  37. configs/dataset/paralleldomain4d_wai/default.yaml +3 -0
  38. configs/dataset/paralleldomain4d_wai/train/default.yaml +26 -0
  39. configs/dataset/paralleldomain4d_wai/val/default.yaml +26 -0
  40. configs/dataset/resolution_options/default.yaml +77 -0
  41. configs/dataset/sailvos3d_wai/default.yaml +3 -0
  42. configs/dataset/sailvos3d_wai/train/default.yaml +26 -0
  43. configs/dataset/sailvos3d_wai/val/default.yaml +26 -0
  44. configs/dataset/scannetpp_wai/default.yaml +4 -0
  45. configs/dataset/scannetpp_wai/test/default.yaml +24 -0
  46. configs/dataset/scannetpp_wai/train/default.yaml +26 -0
  47. configs/dataset/scannetpp_wai/val/default.yaml +26 -0
  48. configs/dataset/spring_wai/default.yaml +3 -0
  49. configs/dataset/spring_wai/train/default.yaml +26 -0
  50. configs/dataset/spring_wai/val/default.yaml +26 -0
configs/calibration_benchmark.yaml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - machine: aws
3
+ - model: default
4
+ - dataset: default
5
+ - _self_
6
+
7
+ output_dir: ${hydra:run.dir}
8
+ root_data_dir: ${machine.root_data_dir}
9
+ mapanything_dataset_metadata_dir: ${machine.mapanything_dataset_metadata_dir}
10
+ root_pretrained_checkpoints_dir: ${machine.root_pretrained_checkpoints_dir}
11
+ root_experiments_dir: ${machine.root_experiments_dir}
12
+ root_uniception_pretrained_checkpoints_dir: ${machine.root_uniception_pretrained_checkpoints_dir}
13
+
14
+ ### Benchmarking args
15
+ seed: 0
16
+ # Disable CUDNN Benchmark (Disable for variable resolution & number of view training)
17
+ disable_cudnn_benchmark: true
18
+ # Batch size for inference (Metrics are computed per multi-view set and averaged, not per batch of multi-view sets)
19
+ batch_size: 20
20
+ # Use mixed precision for inference
21
+ amp: 1
22
+ # Floating point type to use for mixed precision
23
+ amp_dtype: "bf16"
configs/dataset/ase_wai/default.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
configs/dataset/ase_wai/train/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "ASEWAI(
3
+ split='${dataset.ase_wai.train.split}',
4
+ resolution=${dataset.ase_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.ase_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.ase_wai.train.aug_crop},
7
+ transform='${dataset.ase_wai.train.transform}',
8
+ data_norm_type='${dataset.ase_wai.train.data_norm_type}',
9
+ ROOT='${dataset.ase_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.ase_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.ase_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.ase_wai.train.variable_num_views},
13
+ num_views=${dataset.ase_wai.train.num_views},
14
+ covisibility_thres=${dataset.ase_wai.train.covisibility_thres})"
15
+ split: 'train'
16
+ dataset_resolution: ${dataset.resolution_train}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ aug_crop: 16
19
+ transform: 'colorjitter+grayscale+gaublur'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/ase
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.train.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/ase_wai/val/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "ASEWAI(
3
+ split='${dataset.ase_wai.val.split}',
4
+ resolution=${dataset.ase_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.ase_wai.val.principal_point_centered},
6
+ seed=${dataset.ase_wai.val.seed},
7
+ transform='${dataset.ase_wai.val.transform}',
8
+ data_norm_type='${dataset.ase_wai.val.data_norm_type}',
9
+ ROOT='${dataset.ase_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.ase_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.ase_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.ase_wai.val.variable_num_views},
13
+ num_views=${dataset.ase_wai.val.num_views},
14
+ covisibility_thres=${dataset.ase_wai.val.covisibility_thres})"
15
+ split: 'val'
16
+ dataset_resolution: ${dataset.resolution_val_ase}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ seed: 777
19
+ transform: 'imgnorm'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/ase
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.val.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/benchmark_512_eth3d_snpp_tav2.yaml ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 2
6
+
7
+ # Test Resolution
8
+ resolution_test_eth3d: ${dataset.resolution_options.512_1_52_ar}
9
+ resolution_test_scannetpp: ${dataset.resolution_options.512_1_52_ar}
10
+ resolution_test_tav2_wb: ${dataset.resolution_options.512_1_00_ar}
11
+
12
+ # Test Set
13
+ # Sample 10 multi-view sets from each scene
14
+ # ETH3D: 13 scenes
15
+ # ScanNet++V2: 30 scenes
16
+ # TartanAirV2-WB: 5 scenes
17
+ test_dataset:
18
+ "+ 130 @ ${dataset.eth3d_wai.test.dataset_str}
19
+ + 300 @ ${dataset.scannetpp_wai.test.dataset_str}
20
+ + 50 @ ${dataset.tav2_wb_wai.test.dataset_str}"
configs/dataset/benchmark_518_eth3d_snpp_tav2.yaml ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 2
6
+
7
+ # Test Resolution
8
+ resolution_test_eth3d: ${dataset.resolution_options.518_1_52_ar}
9
+ resolution_test_scannetpp: ${dataset.resolution_options.518_1_52_ar}
10
+ resolution_test_tav2_wb: ${dataset.resolution_options.518_1_00_ar}
11
+
12
+ # Test Set
13
+ # Sample 10 multi-view sets from each scene
14
+ # ETH3D: 13 scenes
15
+ # ScanNet++V2: 30 scenes
16
+ # TartanAirV2-WB: 5 scenes
17
+ test_dataset:
18
+ "+ 130 @ ${dataset.eth3d_wai.test.dataset_str}
19
+ + 300 @ ${dataset.scannetpp_wai.test.dataset_str}
20
+ + 50 @ ${dataset.tav2_wb_wai.test.dataset_str}"
configs/dataset/benchmark_sv_calib_518_many_ar_eth3d_snpp_tav2.yaml ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 1
6
+
7
+ # Test Resolution
8
+ resolution_test_eth3d: ${dataset.resolution_options.518_many_ar}
9
+ resolution_test_scannetpp: ${dataset.resolution_options.518_many_ar}
10
+ resolution_test_tav2_wb: ${dataset.resolution_options.518_many_ar}
11
+
12
+ # Test Set
13
+ # Sample 20 frames from each scene
14
+ # ETH3D: 13 scenes
15
+ # ScanNet++V2: 30 scenes
16
+ # TartanAirV2-WB: 5 scenes
17
+ test_dataset:
18
+ "+ 260 @ ${dataset.eth3d_wai.test.dataset_str}
19
+ + 600 @ ${dataset.scannetpp_wai.test.dataset_str}
20
+ + 100 @ ${dataset.tav2_wb_wai.test.dataset_str}"
configs/dataset/blendedmvs_wai/default.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
configs/dataset/blendedmvs_wai/train/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "BlendedMVSWAI(
3
+ split='${dataset.blendedmvs_wai.train.split}',
4
+ resolution=${dataset.blendedmvs_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.blendedmvs_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.blendedmvs_wai.train.aug_crop},
7
+ transform='${dataset.blendedmvs_wai.train.transform}',
8
+ data_norm_type='${dataset.blendedmvs_wai.train.data_norm_type}',
9
+ ROOT='${dataset.blendedmvs_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.blendedmvs_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.blendedmvs_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.blendedmvs_wai.train.variable_num_views},
13
+ num_views=${dataset.blendedmvs_wai.train.num_views},
14
+ covisibility_thres=${dataset.blendedmvs_wai.train.covisibility_thres})"
15
+ split: 'train'
16
+ dataset_resolution: ${dataset.resolution_train}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ aug_crop: 16
19
+ transform: 'colorjitter+grayscale+gaublur'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/blendedmvs
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.train.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/blendedmvs_wai/val/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "BlendedMVSWAI(
3
+ split='${dataset.blendedmvs_wai.val.split}',
4
+ resolution=${dataset.blendedmvs_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.blendedmvs_wai.val.principal_point_centered},
6
+ seed=${dataset.blendedmvs_wai.val.seed},
7
+ transform='${dataset.blendedmvs_wai.val.transform}',
8
+ data_norm_type='${dataset.blendedmvs_wai.val.data_norm_type}',
9
+ ROOT='${dataset.blendedmvs_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.blendedmvs_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.blendedmvs_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.blendedmvs_wai.val.variable_num_views},
13
+ num_views=${dataset.blendedmvs_wai.val.num_views},
14
+ covisibility_thres=${dataset.blendedmvs_wai.val.covisibility_thres})"
15
+ split: 'val'
16
+ dataset_resolution: ${dataset.resolution_val_blendedmvs}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ seed: 777
19
+ transform: 'imgnorm'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/blendedmvs
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.val.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/bmvs_518_many_ar_48ipg_8g.yaml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 4
6
+
7
+ train:
8
+ # If True, the number of views can vary from batch to batch. The maximum number of views is num_views and minimum is 2. (On by default for N-view training)
9
+ variable_num_views: true
10
+
11
+ # Train Resolution
12
+ resolution_train: ${dataset.resolution_options.518_many_ar}
13
+
14
+ # Validation Resolution
15
+ resolution_val_blendedmvs: ${dataset.resolution_options.518_1_33_ar}
16
+
17
+ # Training Set
18
+ train_dataset:
19
+ "+ 140_000 @ ${dataset.blendedmvs_wai.train.dataset_str}"
20
+
21
+ # Validation Set
22
+ test_dataset:
23
+ "+ 4_000 @ ${dataset.blendedmvs_wai.val.dataset_str}"
configs/dataset/default.yaml ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - resolution_options: default
3
+ - ase_wai: default
4
+ - blendedmvs_wai: default
5
+ - dl3dv_wai: default
6
+ - dynamicreplica_wai: default
7
+ - eth3d_wai: default
8
+ - megadepth_wai: default
9
+ - mpsd_wai: default
10
+ - mvs_synth_wai: default
11
+ - paralleldomain4d_wai: default
12
+ - sailvos3d_wai: default
13
+ - scannetpp_wai: default
14
+ - spring_wai: default
15
+ - tav2_wb_wai: default
16
+ - unrealstereo4k_wai: default
17
+
18
+ # Training Set, For example: BlendedMVS(split='train', resolution=(512, 384), transform=...)
19
+ train_dataset: ???
20
+ # Validation Set
21
+ test_dataset: "[null]"
22
+ # Number of workers for dataloader
23
+ num_workers: 12
24
+ # Default resolution for training
25
+ resolution_train: ???
26
+ # Default resolution for validation
27
+ resolution_val: ???
28
+ # Number of views parameter for multi-view datasets
29
+ num_views: 2
30
+ # Use a centered principal point for all images
31
+ principal_point_centered: false
32
+ # Default config for multi-view datasets
33
+ train:
34
+ # If True, the number of views can vary from batch to batch. The maximum number of views is num_views and minimum is 2. (On by default for N-view training)
35
+ variable_num_views: true
36
+ val:
37
+ variable_num_views: false
38
+ test:
39
+ variable_num_views: false
configs/dataset/dl3dv_wai/default.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
configs/dataset/dl3dv_wai/train/default.yaml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "DL3DVWAI(
3
+ split='${dataset.dl3dv_wai.train.split}',
4
+ resolution=${dataset.dl3dv_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.dl3dv_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.dl3dv_wai.train.aug_crop},
7
+ transform='${dataset.dl3dv_wai.train.transform}',
8
+ data_norm_type='${dataset.dl3dv_wai.train.data_norm_type}',
9
+ ROOT='${dataset.dl3dv_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.dl3dv_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.dl3dv_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.dl3dv_wai.train.variable_num_views},
13
+ num_views=${dataset.dl3dv_wai.train.num_views},
14
+ covisibility_thres=${dataset.dl3dv_wai.train.covisibility_thres},
15
+ mvs_confidence_filter_thres=${dataset.dl3dv_wai.train.mvs_confidence_filter_thres})"
16
+ split: 'train'
17
+ dataset_resolution: ${dataset.resolution_train}
18
+ principal_point_centered: ${dataset.principal_point_centered}
19
+ aug_crop: 16
20
+ transform: 'colorjitter+grayscale+gaublur'
21
+ data_norm_type: ${model.data_norm_type}
22
+ ROOT: ${root_data_dir}/dl3dv
23
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
24
+ overfit_num_sets: null
25
+ variable_num_views: ${dataset.train.variable_num_views}
26
+ num_views: ${dataset.num_views}
27
+ covisibility_thres: 0.25
28
+ mvs_confidence_filter_thres: 0.25
configs/dataset/dl3dv_wai/val/default.yaml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "DL3DVWAI(
3
+ split='${dataset.dl3dv_wai.val.split}',
4
+ resolution=${dataset.dl3dv_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.dl3dv_wai.val.principal_point_centered},
6
+ seed=${dataset.dl3dv_wai.val.seed},
7
+ transform='${dataset.dl3dv_wai.val.transform}',
8
+ data_norm_type='${dataset.dl3dv_wai.val.data_norm_type}',
9
+ ROOT='${dataset.dl3dv_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.dl3dv_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.dl3dv_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.dl3dv_wai.val.variable_num_views},
13
+ num_views=${dataset.dl3dv_wai.val.num_views},
14
+ covisibility_thres=${dataset.dl3dv_wai.val.covisibility_thres},
15
+ mvs_confidence_filter_thres=${dataset.dl3dv_wai.val.mvs_confidence_filter_thres})"
16
+ split: 'val'
17
+ dataset_resolution: ${dataset.resolution_val_dl3dv}
18
+ principal_point_centered: ${dataset.principal_point_centered}
19
+ seed: 777
20
+ transform: 'imgnorm'
21
+ data_norm_type: ${model.data_norm_type}
22
+ ROOT: ${root_data_dir}/dl3dv
23
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
24
+ overfit_num_sets: null
25
+ variable_num_views: ${dataset.val.variable_num_views}
26
+ num_views: ${dataset.num_views}
27
+ covisibility_thres: 0.25
28
+ mvs_confidence_filter_thres: 0.25
configs/dataset/dynamicreplica_wai/default.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
configs/dataset/dynamicreplica_wai/train/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "DynamicReplicaWAI(
3
+ split='${dataset.dynamicreplica_wai.train.split}',
4
+ resolution=${dataset.dynamicreplica_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.dynamicreplica_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.dynamicreplica_wai.train.aug_crop},
7
+ transform='${dataset.dynamicreplica_wai.train.transform}',
8
+ data_norm_type='${dataset.dynamicreplica_wai.train.data_norm_type}',
9
+ ROOT='${dataset.dynamicreplica_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.dynamicreplica_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.dynamicreplica_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.dynamicreplica_wai.train.variable_num_views},
13
+ num_views=${dataset.dynamicreplica_wai.train.num_views},
14
+ covisibility_thres=${dataset.dynamicreplica_wai.train.covisibility_thres})"
15
+ split: 'train'
16
+ dataset_resolution: ${dataset.resolution_train}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ aug_crop: 16
19
+ transform: 'colorjitter+grayscale+gaublur'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/dynamicreplica
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.train.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/dynamicreplica_wai/val/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "DynamicReplicaWAI(
3
+ split='${dataset.dynamicreplica_wai.val.split}',
4
+ resolution=${dataset.dynamicreplica_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.dynamicreplica_wai.val.principal_point_centered},
6
+ seed=${dataset.dynamicreplica_wai.val.seed},
7
+ transform='${dataset.dynamicreplica_wai.val.transform}',
8
+ data_norm_type='${dataset.dynamicreplica_wai.val.data_norm_type}',
9
+ ROOT='${dataset.dynamicreplica_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.dynamicreplica_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.dynamicreplica_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.dynamicreplica_wai.val.variable_num_views},
13
+ num_views=${dataset.dynamicreplica_wai.val.num_views},
14
+ covisibility_thres=${dataset.dynamicreplica_wai.val.covisibility_thres})"
15
+ split: 'val'
16
+ dataset_resolution: ${dataset.resolution_val_dynamicreplica}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ seed: 777
19
+ transform: 'imgnorm'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/dynamicreplica
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.val.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/eth3d_wai/default.yaml ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ defaults:
2
+ - test: default
configs/dataset/eth3d_wai/test/default.yaml ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "ETH3DWAI(
3
+ resolution=${dataset.eth3d_wai.test.dataset_resolution},
4
+ principal_point_centered=${dataset.eth3d_wai.test.principal_point_centered},
5
+ seed=${dataset.eth3d_wai.test.seed},
6
+ transform='${dataset.eth3d_wai.test.transform}',
7
+ data_norm_type='${dataset.eth3d_wai.test.data_norm_type}',
8
+ ROOT='${dataset.eth3d_wai.test.ROOT}',
9
+ dataset_metadata_dir='${dataset.eth3d_wai.test.dataset_metadata_dir}',
10
+ variable_num_views=${dataset.eth3d_wai.test.variable_num_views},
11
+ num_views=${dataset.eth3d_wai.test.num_views},
12
+ covisibility_thres=${dataset.eth3d_wai.test.covisibility_thres})"
13
+ dataset_resolution: ${dataset.resolution_test_eth3d}
14
+ principal_point_centered: ${dataset.principal_point_centered}
15
+ seed: 777
16
+ transform: 'imgnorm'
17
+ data_norm_type: ${model.data_norm_type}
18
+ ROOT: ${root_data_dir}/eth3d
19
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
20
+ variable_num_views: ${dataset.test.variable_num_views}
21
+ num_views: ${dataset.num_views}
22
+ covisibility_thres: 0.025
configs/dataset/megadepth_wai/default.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
configs/dataset/megadepth_wai/train/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "MegaDepthWAI(
3
+ split='${dataset.megadepth_wai.train.split}',
4
+ resolution=${dataset.megadepth_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.megadepth_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.megadepth_wai.train.aug_crop},
7
+ transform='${dataset.megadepth_wai.train.transform}',
8
+ data_norm_type='${dataset.megadepth_wai.train.data_norm_type}',
9
+ ROOT='${dataset.megadepth_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.megadepth_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.megadepth_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.megadepth_wai.train.variable_num_views},
13
+ num_views=${dataset.megadepth_wai.train.num_views},
14
+ covisibility_thres=${dataset.megadepth_wai.train.covisibility_thres})"
15
+ split: 'train'
16
+ dataset_resolution: ${dataset.resolution_train}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ aug_crop: 16
19
+ transform: 'colorjitter+grayscale+gaublur'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/megadepth
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.train.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/megadepth_wai/val/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "MegaDepthWAI(
3
+ split='${dataset.megadepth_wai.val.split}',
4
+ resolution=${dataset.megadepth_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.megadepth_wai.val.principal_point_centered},
6
+ seed=${dataset.megadepth_wai.val.seed},
7
+ transform='${dataset.megadepth_wai.val.transform}',
8
+ data_norm_type='${dataset.megadepth_wai.val.data_norm_type}',
9
+ ROOT='${dataset.megadepth_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.megadepth_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.megadepth_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.megadepth_wai.val.variable_num_views},
13
+ num_views=${dataset.megadepth_wai.val.num_views},
14
+ covisibility_thres=${dataset.megadepth_wai.val.covisibility_thres})"
15
+ split: 'val'
16
+ dataset_resolution: ${dataset.resolution_val_megadepth}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ seed: 777
19
+ transform: 'imgnorm'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/megadepth
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.val.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/megatrain_13d_512_many_ar_24ipg_16g.yaml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 4
6
+
7
+ train:
8
+ # If True, the number of views can vary from batch to batch. The maximum number of views is num_views and minimum is 2. (On by default for N-view training)
9
+ variable_num_views: true
10
+
11
+ # Train Resolution
12
+ resolution_train: ${dataset.resolution_options.512_many_ar}
13
+
14
+ # Validation Resolution
15
+ resolution_val_ase: ${dataset.resolution_options.512_1_00_ar}
16
+ resolution_val_blendedmvs: ${dataset.resolution_options.512_1_33_ar}
17
+ resolution_val_dl3dv: ${dataset.resolution_options.512_1_77_ar}
18
+ resolution_val_dynamicreplica: ${dataset.resolution_options.512_1_77_ar}
19
+ resolution_val_megadepth: ${dataset.resolution_options.512_1_52_ar}
20
+ resolution_val_mpsd: ${dataset.resolution_options.512_1_77_ar}
21
+ resolution_val_mvs_synth: ${dataset.resolution_options.512_1_77_ar}
22
+ resolution_val_paralleldomain4d: ${dataset.resolution_options.512_1_33_ar}
23
+ resolution_val_sailvos3d: ${dataset.resolution_options.512_1_52_ar}
24
+ resolution_val_scannetpp: ${dataset.resolution_options.512_1_52_ar}
25
+ resolution_val_spring: ${dataset.resolution_options.512_1_77_ar}
26
+ resolution_val_tav2_wb: ${dataset.resolution_options.512_1_00_ar}
27
+ resolution_val_unrealstereo4k: ${dataset.resolution_options.512_1_77_ar}
28
+
29
+ # Training Set
30
+ train_dataset:
31
+ "+ 52_500 @ ${dataset.ase_wai.train.dataset_str}
32
+ + 52_500 @ ${dataset.blendedmvs_wai.train.dataset_str}
33
+ + 52_500 @ ${dataset.dl3dv_wai.train.dataset_str}
34
+ + 40_000 @ ${dataset.dynamicreplica_wai.train.dataset_str}
35
+ + 52_500 @ ${dataset.megadepth_wai.train.dataset_str}
36
+ + 52_500 @ ${dataset.mpsd_wai.train.dataset_str}
37
+ + 52_500 @ ${dataset.mvs_synth_wai.train.dataset_str}
38
+ + 52_500 @ ${dataset.paralleldomain4d_wai.train.dataset_str}
39
+ + 52_500 @ ${dataset.sailvos3d_wai.train.dataset_str}
40
+ + 52_500 @ ${dataset.scannetpp_wai.train.dataset_str}
41
+ + 2_000 @ ${dataset.spring_wai.train.dataset_str}
42
+ + 52_500 @ ${dataset.tav2_wb_wai.train.dataset_str}
43
+ + 5_500 @ ${dataset.unrealstereo4k_wai.train.dataset_str}"
44
+
45
+ # Validation Set
46
+ test_dataset:
47
+ "+ 4_000 @ ${dataset.ase_wai.val.dataset_str}
48
+ + 4_000 @ ${dataset.blendedmvs_wai.val.dataset_str}
49
+ + 4_000 @ ${dataset.dl3dv_wai.val.dataset_str}
50
+ + 4_000 @ ${dataset.dynamicreplica_wai.val.dataset_str}
51
+ + 4_000 @ ${dataset.megadepth_wai.val.dataset_str}
52
+ + 4_000 @ ${dataset.mpsd_wai.val.dataset_str}
53
+ + 4_000 @ ${dataset.mvs_synth_wai.val.dataset_str}
54
+ + 4_000 @ ${dataset.paralleldomain4d_wai.val.dataset_str}
55
+ + 4_000 @ ${dataset.sailvos3d_wai.val.dataset_str}
56
+ + 4_000 @ ${dataset.scannetpp_wai.val.dataset_str}
57
+ + 500 @ ${dataset.spring_wai.val.dataset_str}
58
+ + 4_000 @ ${dataset.tav2_wb_wai.val.dataset_str}
59
+ + 500 @ ${dataset.unrealstereo4k_wai.val.dataset_str}"
configs/dataset/megatrain_13d_518_many_ar_24ipg_16g.yaml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 4
6
+
7
+ train:
8
+ # If True, the number of views can vary from batch to batch. The maximum number of views is num_views and minimum is 2. (On by default for N-view training)
9
+ variable_num_views: true
10
+
11
+ # Train Resolution
12
+ resolution_train: ${dataset.resolution_options.518_many_ar}
13
+
14
+ # Validation Resolution
15
+ resolution_val_ase: ${dataset.resolution_options.518_1_00_ar}
16
+ resolution_val_blendedmvs: ${dataset.resolution_options.518_1_33_ar}
17
+ resolution_val_dl3dv: ${dataset.resolution_options.518_1_77_ar}
18
+ resolution_val_dynamicreplica: ${dataset.resolution_options.518_1_77_ar}
19
+ resolution_val_megadepth: ${dataset.resolution_options.518_1_52_ar}
20
+ resolution_val_mpsd: ${dataset.resolution_options.518_1_77_ar}
21
+ resolution_val_mvs_synth: ${dataset.resolution_options.518_1_77_ar}
22
+ resolution_val_paralleldomain4d: ${dataset.resolution_options.518_1_33_ar}
23
+ resolution_val_sailvos3d: ${dataset.resolution_options.518_1_52_ar}
24
+ resolution_val_scannetpp: ${dataset.resolution_options.518_1_52_ar}
25
+ resolution_val_spring: ${dataset.resolution_options.518_1_77_ar}
26
+ resolution_val_tav2_wb: ${dataset.resolution_options.518_1_00_ar}
27
+ resolution_val_unrealstereo4k: ${dataset.resolution_options.518_1_77_ar}
28
+
29
+ # Training Set
30
+ train_dataset:
31
+ "+ 52_500 @ ${dataset.ase_wai.train.dataset_str}
32
+ + 52_500 @ ${dataset.blendedmvs_wai.train.dataset_str}
33
+ + 52_500 @ ${dataset.dl3dv_wai.train.dataset_str}
34
+ + 40_000 @ ${dataset.dynamicreplica_wai.train.dataset_str}
35
+ + 52_500 @ ${dataset.megadepth_wai.train.dataset_str}
36
+ + 52_500 @ ${dataset.mpsd_wai.train.dataset_str}
37
+ + 52_500 @ ${dataset.mvs_synth_wai.train.dataset_str}
38
+ + 52_500 @ ${dataset.paralleldomain4d_wai.train.dataset_str}
39
+ + 52_500 @ ${dataset.sailvos3d_wai.train.dataset_str}
40
+ + 52_500 @ ${dataset.scannetpp_wai.train.dataset_str}
41
+ + 2_000 @ ${dataset.spring_wai.train.dataset_str}
42
+ + 52_500 @ ${dataset.tav2_wb_wai.train.dataset_str}
43
+ + 5_500 @ ${dataset.unrealstereo4k_wai.train.dataset_str}"
44
+
45
+ # Validation Set
46
+ test_dataset:
47
+ "+ 4_000 @ ${dataset.ase_wai.val.dataset_str}
48
+ + 4_000 @ ${dataset.blendedmvs_wai.val.dataset_str}
49
+ + 4_000 @ ${dataset.dl3dv_wai.val.dataset_str}
50
+ + 4_000 @ ${dataset.dynamicreplica_wai.val.dataset_str}
51
+ + 4_000 @ ${dataset.megadepth_wai.val.dataset_str}
52
+ + 4_000 @ ${dataset.mpsd_wai.val.dataset_str}
53
+ + 4_000 @ ${dataset.mvs_synth_wai.val.dataset_str}
54
+ + 4_000 @ ${dataset.paralleldomain4d_wai.val.dataset_str}
55
+ + 4_000 @ ${dataset.sailvos3d_wai.val.dataset_str}
56
+ + 4_000 @ ${dataset.scannetpp_wai.val.dataset_str}
57
+ + 500 @ ${dataset.spring_wai.val.dataset_str}
58
+ + 4_000 @ ${dataset.tav2_wb_wai.val.dataset_str}
59
+ + 500 @ ${dataset.unrealstereo4k_wai.val.dataset_str}"
configs/dataset/megatrain_13d_518_many_ar_24ipg_8g.yaml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 4
6
+
7
+ train:
8
+ # If True, the number of views can vary from batch to batch. The maximum number of views is num_views and minimum is 2. (On by default for N-view training)
9
+ variable_num_views: true
10
+
11
+ # Train Resolution
12
+ resolution_train: ${dataset.resolution_options.518_many_ar}
13
+
14
+ # Validation Resolution
15
+ resolution_val_ase: ${dataset.resolution_options.518_1_00_ar}
16
+ resolution_val_blendedmvs: ${dataset.resolution_options.518_1_33_ar}
17
+ resolution_val_dl3dv: ${dataset.resolution_options.518_1_77_ar}
18
+ resolution_val_dynamicreplica: ${dataset.resolution_options.518_1_77_ar}
19
+ resolution_val_megadepth: ${dataset.resolution_options.518_1_52_ar}
20
+ resolution_val_mpsd: ${dataset.resolution_options.518_1_77_ar}
21
+ resolution_val_mvs_synth: ${dataset.resolution_options.518_1_77_ar}
22
+ resolution_val_paralleldomain4d: ${dataset.resolution_options.518_1_33_ar}
23
+ resolution_val_sailvos3d: ${dataset.resolution_options.518_1_52_ar}
24
+ resolution_val_scannetpp: ${dataset.resolution_options.518_1_52_ar}
25
+ resolution_val_spring: ${dataset.resolution_options.518_1_77_ar}
26
+ resolution_val_tav2_wb: ${dataset.resolution_options.518_1_00_ar}
27
+ resolution_val_unrealstereo4k: ${dataset.resolution_options.518_1_77_ar}
28
+
29
+ # Training Set
30
+ train_dataset:
31
+ "+ 26_250 @ ${dataset.ase_wai.train.dataset_str}
32
+ + 26_250 @ ${dataset.blendedmvs_wai.train.dataset_str}
33
+ + 26_250 @ ${dataset.dl3dv_wai.train.dataset_str}
34
+ + 20_000 @ ${dataset.dynamicreplica_wai.train.dataset_str}
35
+ + 26_250 @ ${dataset.megadepth_wai.train.dataset_str}
36
+ + 26_250 @ ${dataset.mpsd_wai.train.dataset_str}
37
+ + 26_250 @ ${dataset.mvs_synth_wai.train.dataset_str}
38
+ + 26_250 @ ${dataset.paralleldomain4d_wai.train.dataset_str}
39
+ + 26_250 @ ${dataset.sailvos3d_wai.train.dataset_str}
40
+ + 26_250 @ ${dataset.scannetpp_wai.train.dataset_str}
41
+ + 1_000 @ ${dataset.spring_wai.train.dataset_str}
42
+ + 26_250 @ ${dataset.tav2_wb_wai.train.dataset_str}
43
+ + 2_750 @ ${dataset.unrealstereo4k_wai.train.dataset_str}"
44
+
45
+ # Validation Set
46
+ test_dataset:
47
+ "+ 4_000 @ ${dataset.ase_wai.val.dataset_str}
48
+ + 4_000 @ ${dataset.blendedmvs_wai.val.dataset_str}
49
+ + 4_000 @ ${dataset.dl3dv_wai.val.dataset_str}
50
+ + 4_000 @ ${dataset.dynamicreplica_wai.val.dataset_str}
51
+ + 4_000 @ ${dataset.megadepth_wai.val.dataset_str}
52
+ + 4_000 @ ${dataset.mpsd_wai.val.dataset_str}
53
+ + 4_000 @ ${dataset.mvs_synth_wai.val.dataset_str}
54
+ + 4_000 @ ${dataset.paralleldomain4d_wai.val.dataset_str}
55
+ + 4_000 @ ${dataset.sailvos3d_wai.val.dataset_str}
56
+ + 4_000 @ ${dataset.scannetpp_wai.val.dataset_str}
57
+ + 500 @ ${dataset.spring_wai.val.dataset_str}
58
+ + 4_000 @ ${dataset.tav2_wb_wai.val.dataset_str}
59
+ + 500 @ ${dataset.unrealstereo4k_wai.val.dataset_str}"
configs/dataset/megatrain_13d_518_many_ar_48ipg_64g.yaml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 4
6
+
7
+ train:
8
+ # If True, the number of views can vary from batch to batch. The maximum number of views is num_views and minimum is 2. (On by default for N-view training)
9
+ variable_num_views: true
10
+
11
+ # Train Resolution
12
+ resolution_train: ${dataset.resolution_options.518_many_ar}
13
+
14
+ # Validation Resolution
15
+ resolution_val_ase: ${dataset.resolution_options.518_1_00_ar}
16
+ resolution_val_blendedmvs: ${dataset.resolution_options.518_1_33_ar}
17
+ resolution_val_dl3dv: ${dataset.resolution_options.518_1_77_ar}
18
+ resolution_val_dynamicreplica: ${dataset.resolution_options.518_1_77_ar}
19
+ resolution_val_megadepth: ${dataset.resolution_options.518_1_52_ar}
20
+ resolution_val_mpsd: ${dataset.resolution_options.518_1_77_ar}
21
+ resolution_val_mvs_synth: ${dataset.resolution_options.518_1_77_ar}
22
+ resolution_val_paralleldomain4d: ${dataset.resolution_options.518_1_33_ar}
23
+ resolution_val_sailvos3d: ${dataset.resolution_options.518_1_52_ar}
24
+ resolution_val_scannetpp: ${dataset.resolution_options.518_1_52_ar}
25
+ resolution_val_spring: ${dataset.resolution_options.518_1_77_ar}
26
+ resolution_val_tav2_wb: ${dataset.resolution_options.518_1_00_ar}
27
+ resolution_val_unrealstereo4k: ${dataset.resolution_options.518_1_77_ar}
28
+
29
+ # Training Set
30
+ train_dataset:
31
+ "+ 420_000 @ ${dataset.ase_wai.train.dataset_str}
32
+ + 420_000 @ ${dataset.blendedmvs_wai.train.dataset_str}
33
+ + 420_000 @ ${dataset.dl3dv_wai.train.dataset_str}
34
+ + 320_000 @ ${dataset.dynamicreplica_wai.train.dataset_str}
35
+ + 420_000 @ ${dataset.megadepth_wai.train.dataset_str}
36
+ + 420_000 @ ${dataset.mpsd_wai.train.dataset_str}
37
+ + 420_000 @ ${dataset.mvs_synth_wai.train.dataset_str}
38
+ + 420_000 @ ${dataset.paralleldomain4d_wai.train.dataset_str}
39
+ + 420_000 @ ${dataset.sailvos3d_wai.train.dataset_str}
40
+ + 420_000 @ ${dataset.scannetpp_wai.train.dataset_str}
41
+ + 16_000 @ ${dataset.spring_wai.train.dataset_str}
42
+ + 420_000 @ ${dataset.tav2_wb_wai.train.dataset_str}
43
+ + 44_000 @ ${dataset.unrealstereo4k_wai.train.dataset_str}"
44
+
45
+ # Validation Set
46
+ test_dataset:
47
+ "+ 4_000 @ ${dataset.ase_wai.val.dataset_str}
48
+ + 4_000 @ ${dataset.blendedmvs_wai.val.dataset_str}
49
+ + 4_000 @ ${dataset.dl3dv_wai.val.dataset_str}
50
+ + 4_000 @ ${dataset.dynamicreplica_wai.val.dataset_str}
51
+ + 4_000 @ ${dataset.megadepth_wai.val.dataset_str}
52
+ + 4_000 @ ${dataset.mpsd_wai.val.dataset_str}
53
+ + 4_000 @ ${dataset.mvs_synth_wai.val.dataset_str}
54
+ + 4_000 @ ${dataset.paralleldomain4d_wai.val.dataset_str}
55
+ + 4_000 @ ${dataset.sailvos3d_wai.val.dataset_str}
56
+ + 4_000 @ ${dataset.scannetpp_wai.val.dataset_str}
57
+ + 500 @ ${dataset.spring_wai.val.dataset_str}
58
+ + 4_000 @ ${dataset.tav2_wb_wai.val.dataset_str}
59
+ + 500 @ ${dataset.unrealstereo4k_wai.val.dataset_str}"
configs/dataset/megatrain_13d_518_many_ar_48ipg_8g_mono.yaml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 1
6
+
7
+ train:
8
+ # If True, the number of views can vary from batch to batch. The maximum number of views is num_views and minimum is 2. (On by default for N-view training)
9
+ variable_num_views: true
10
+
11
+ # Train Resolution
12
+ resolution_train: ${dataset.resolution_options.518_many_ar}
13
+
14
+ # Validation Resolution
15
+ resolution_val_ase: ${dataset.resolution_options.518_1_00_ar}
16
+ resolution_val_blendedmvs: ${dataset.resolution_options.518_1_33_ar}
17
+ resolution_val_dl3dv: ${dataset.resolution_options.518_1_77_ar}
18
+ resolution_val_dynamicreplica: ${dataset.resolution_options.518_1_77_ar}
19
+ resolution_val_megadepth: ${dataset.resolution_options.518_1_52_ar}
20
+ resolution_val_mpsd: ${dataset.resolution_options.518_1_77_ar}
21
+ resolution_val_mvs_synth: ${dataset.resolution_options.518_1_77_ar}
22
+ resolution_val_paralleldomain4d: ${dataset.resolution_options.518_1_33_ar}
23
+ resolution_val_sailvos3d: ${dataset.resolution_options.518_1_52_ar}
24
+ resolution_val_scannetpp: ${dataset.resolution_options.518_1_52_ar}
25
+ resolution_val_spring: ${dataset.resolution_options.518_1_77_ar}
26
+ resolution_val_tav2_wb: ${dataset.resolution_options.518_1_00_ar}
27
+ resolution_val_unrealstereo4k: ${dataset.resolution_options.518_1_77_ar}
28
+
29
+ # Training Set
30
+ train_dataset:
31
+ "+ 105_000 @ ${dataset.ase_wai.train.dataset_str}
32
+ + 105_000 @ ${dataset.blendedmvs_wai.train.dataset_str}
33
+ + 105_000 @ ${dataset.dl3dv_wai.train.dataset_str}
34
+ + 80_000 @ ${dataset.dynamicreplica_wai.train.dataset_str}
35
+ + 105_000 @ ${dataset.megadepth_wai.train.dataset_str}
36
+ + 105_000 @ ${dataset.mpsd_wai.train.dataset_str}
37
+ + 105_000 @ ${dataset.mvs_synth_wai.train.dataset_str}
38
+ + 105_000 @ ${dataset.paralleldomain4d_wai.train.dataset_str}
39
+ + 105_000 @ ${dataset.sailvos3d_wai.train.dataset_str}
40
+ + 105_000 @ ${dataset.scannetpp_wai.train.dataset_str}
41
+ + 4_000 @ ${dataset.spring_wai.train.dataset_str}
42
+ + 105_000 @ ${dataset.tav2_wb_wai.train.dataset_str}
43
+ + 11_000 @ ${dataset.unrealstereo4k_wai.train.dataset_str}"
44
+
45
+ # Validation Set
46
+ test_dataset:
47
+ "+ 4_000 @ ${dataset.ase_wai.val.dataset_str}
48
+ + 4_000 @ ${dataset.blendedmvs_wai.val.dataset_str}
49
+ + 4_000 @ ${dataset.dl3dv_wai.val.dataset_str}
50
+ + 4_000 @ ${dataset.dynamicreplica_wai.val.dataset_str}
51
+ + 4_000 @ ${dataset.megadepth_wai.val.dataset_str}
52
+ + 4_000 @ ${dataset.mpsd_wai.val.dataset_str}
53
+ + 4_000 @ ${dataset.mvs_synth_wai.val.dataset_str}
54
+ + 4_000 @ ${dataset.paralleldomain4d_wai.val.dataset_str}
55
+ + 4_000 @ ${dataset.sailvos3d_wai.val.dataset_str}
56
+ + 4_000 @ ${dataset.scannetpp_wai.val.dataset_str}
57
+ + 500 @ ${dataset.spring_wai.val.dataset_str}
58
+ + 4_000 @ ${dataset.tav2_wb_wai.val.dataset_str}
59
+ + 500 @ ${dataset.unrealstereo4k_wai.val.dataset_str}"
configs/dataset/megatrain_6d_518_many_ar_48ipg_64g.yaml ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 4
6
+
7
+ train:
8
+ # If True, the number of views can vary from batch to batch. The maximum number of views is num_views and minimum is 2. (On by default for N-view training)
9
+ variable_num_views: true
10
+
11
+ # Train Resolution
12
+ resolution_train: ${dataset.resolution_options.518_many_ar}
13
+
14
+ # Validation Resolution
15
+ resolution_val_blendedmvs: ${dataset.resolution_options.518_1_33_ar}
16
+ resolution_val_mpsd: ${dataset.resolution_options.518_1_77_ar}
17
+ resolution_val_scannetpp: ${dataset.resolution_options.518_1_52_ar}
18
+ resolution_val_spring: ${dataset.resolution_options.518_1_77_ar}
19
+ resolution_val_tav2_wb: ${dataset.resolution_options.518_1_00_ar}
20
+ resolution_val_unrealstereo4k: ${dataset.resolution_options.518_1_77_ar}
21
+
22
+ # Training Set
23
+ train_dataset:
24
+ "+ 1_120_000 @ ${dataset.blendedmvs_wai.train.dataset_str}
25
+ + 1_120_000 @ ${dataset.mpsd_wai.train.dataset_str}
26
+ + 1_120_000 @ ${dataset.scannetpp_wai.train.dataset_str}
27
+ + 44_000 @ ${dataset.spring_wai.train.dataset_str}
28
+ + 1_120_000 @ ${dataset.tav2_wb_wai.train.dataset_str}
29
+ + 116_000 @ ${dataset.unrealstereo4k_wai.train.dataset_str}"
30
+
31
+ # Validation Set
32
+ test_dataset:
33
+ "+ 4_000 @ ${dataset.blendedmvs_wai.val.dataset_str}
34
+ + 4_000 @ ${dataset.mpsd_wai.val.dataset_str}
35
+ + 4_000 @ ${dataset.scannetpp_wai.val.dataset_str}
36
+ + 500 @ ${dataset.spring_wai.val.dataset_str}
37
+ + 4_000 @ ${dataset.tav2_wb_wai.val.dataset_str}
38
+ + 500 @ ${dataset.unrealstereo4k_wai.val.dataset_str}"
configs/dataset/megatrain_6d_518_many_ar_48ipg_8g.yaml ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ defaults:
2
+ - default
3
+
4
+ # Number of views parameter for the multi-view datasets
5
+ num_views: 4
6
+
7
+ train:
8
+ # If True, the number of views can vary from batch to batch. The maximum number of views is num_views and minimum is 2. (On by default for N-view training)
9
+ variable_num_views: true
10
+
11
+ # Train Resolution
12
+ resolution_train: ${dataset.resolution_options.518_many_ar}
13
+
14
+ # Validation Resolution
15
+ resolution_val_blendedmvs: ${dataset.resolution_options.518_1_33_ar}
16
+ resolution_val_mpsd: ${dataset.resolution_options.518_1_77_ar}
17
+ resolution_val_scannetpp: ${dataset.resolution_options.518_1_52_ar}
18
+ resolution_val_spring: ${dataset.resolution_options.518_1_77_ar}
19
+ resolution_val_tav2_wb: ${dataset.resolution_options.518_1_00_ar}
20
+ resolution_val_unrealstereo4k: ${dataset.resolution_options.518_1_77_ar}
21
+
22
+ # Training Set
23
+ train_dataset:
24
+ "+ 140_000 @ ${dataset.blendedmvs_wai.train.dataset_str}
25
+ + 140_000 @ ${dataset.mpsd_wai.train.dataset_str}
26
+ + 140_000 @ ${dataset.scannetpp_wai.train.dataset_str}
27
+ + 5_500 @ ${dataset.spring_wai.train.dataset_str}
28
+ + 140_000 @ ${dataset.tav2_wb_wai.train.dataset_str}
29
+ + 14_500 @ ${dataset.unrealstereo4k_wai.train.dataset_str}"
30
+
31
+ # Validation Set
32
+ test_dataset:
33
+ "+ 4_000 @ ${dataset.blendedmvs_wai.val.dataset_str}
34
+ + 4_000 @ ${dataset.mpsd_wai.val.dataset_str}
35
+ + 4_000 @ ${dataset.scannetpp_wai.val.dataset_str}
36
+ + 500 @ ${dataset.spring_wai.val.dataset_str}
37
+ + 4_000 @ ${dataset.tav2_wb_wai.val.dataset_str}
38
+ + 500 @ ${dataset.unrealstereo4k_wai.val.dataset_str}"
configs/dataset/mpsd_wai/default.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
configs/dataset/mpsd_wai/train/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "MPSDWAI(
3
+ split='${dataset.mpsd_wai.train.split}',
4
+ resolution=${dataset.mpsd_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.mpsd_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.mpsd_wai.train.aug_crop},
7
+ transform='${dataset.mpsd_wai.train.transform}',
8
+ data_norm_type='${dataset.mpsd_wai.train.data_norm_type}',
9
+ ROOT='${dataset.mpsd_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.mpsd_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.mpsd_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.mpsd_wai.train.variable_num_views},
13
+ num_views=${dataset.mpsd_wai.train.num_views},
14
+ covisibility_thres=${dataset.mpsd_wai.train.covisibility_thres})"
15
+ split: 'train'
16
+ dataset_resolution: ${dataset.resolution_train}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ aug_crop: 16
19
+ transform: 'colorjitter+grayscale+gaublur'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/mpsd
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.train.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.15
configs/dataset/mpsd_wai/val/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "MPSDWAI(
3
+ split='${dataset.mpsd_wai.val.split}',
4
+ resolution=${dataset.mpsd_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.mpsd_wai.val.principal_point_centered},
6
+ seed=${dataset.mpsd_wai.val.seed},
7
+ transform='${dataset.mpsd_wai.val.transform}',
8
+ data_norm_type='${dataset.mpsd_wai.val.data_norm_type}',
9
+ ROOT='${dataset.mpsd_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.mpsd_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.mpsd_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.mpsd_wai.val.variable_num_views},
13
+ num_views=${dataset.mpsd_wai.val.num_views},
14
+ covisibility_thres=${dataset.mpsd_wai.val.covisibility_thres})"
15
+ split: 'val'
16
+ dataset_resolution: ${dataset.resolution_val_mpsd}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ seed: 777
19
+ transform: 'imgnorm'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/mpsd
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.val.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.15
configs/dataset/mvs_synth_wai/default.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
configs/dataset/mvs_synth_wai/train/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "MVSSynthWAI(
3
+ split='${dataset.mvs_synth_wai.train.split}',
4
+ resolution=${dataset.mvs_synth_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.mvs_synth_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.mvs_synth_wai.train.aug_crop},
7
+ transform='${dataset.mvs_synth_wai.train.transform}',
8
+ data_norm_type='${dataset.mvs_synth_wai.train.data_norm_type}',
9
+ ROOT='${dataset.mvs_synth_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.mvs_synth_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.mvs_synth_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.mvs_synth_wai.train.variable_num_views},
13
+ num_views=${dataset.mvs_synth_wai.train.num_views},
14
+ covisibility_thres=${dataset.mvs_synth_wai.train.covisibility_thres})"
15
+ split: 'train'
16
+ dataset_resolution: ${dataset.resolution_train}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ aug_crop: 16
19
+ transform: 'colorjitter+grayscale+gaublur'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/mvs_synth
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.train.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/mvs_synth_wai/val/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "MVSSynthWAI(
3
+ split='${dataset.mvs_synth_wai.val.split}',
4
+ resolution=${dataset.mvs_synth_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.mvs_synth_wai.val.principal_point_centered},
6
+ seed=${dataset.mvs_synth_wai.val.seed},
7
+ transform='${dataset.mvs_synth_wai.val.transform}',
8
+ data_norm_type='${dataset.mvs_synth_wai.val.data_norm_type}',
9
+ ROOT='${dataset.mvs_synth_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.mvs_synth_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.mvs_synth_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.mvs_synth_wai.val.variable_num_views},
13
+ num_views=${dataset.mvs_synth_wai.val.num_views},
14
+ covisibility_thres=${dataset.mvs_synth_wai.val.covisibility_thres})"
15
+ split: 'val'
16
+ dataset_resolution: ${dataset.resolution_val_mvs_synth}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ seed: 777
19
+ transform: 'imgnorm'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/mvs_synth
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.val.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/paralleldomain4d_wai/default.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
configs/dataset/paralleldomain4d_wai/train/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "ParallelDomain4DWAI(
3
+ split='${dataset.paralleldomain4d_wai.train.split}',
4
+ resolution=${dataset.paralleldomain4d_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.paralleldomain4d_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.paralleldomain4d_wai.train.aug_crop},
7
+ transform='${dataset.paralleldomain4d_wai.train.transform}',
8
+ data_norm_type='${dataset.paralleldomain4d_wai.train.data_norm_type}',
9
+ ROOT='${dataset.paralleldomain4d_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.paralleldomain4d_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.paralleldomain4d_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.paralleldomain4d_wai.train.variable_num_views},
13
+ num_views=${dataset.paralleldomain4d_wai.train.num_views},
14
+ covisibility_thres=${dataset.paralleldomain4d_wai.train.covisibility_thres})"
15
+ split: 'train'
16
+ dataset_resolution: ${dataset.resolution_train}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ aug_crop: 16
19
+ transform: 'colorjitter+grayscale+gaublur'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/paralleldomain4d
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.train.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/paralleldomain4d_wai/val/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "ParallelDomain4DWAI(
3
+ split='${dataset.paralleldomain4d_wai.val.split}',
4
+ resolution=${dataset.paralleldomain4d_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.paralleldomain4d_wai.val.principal_point_centered},
6
+ seed=${dataset.paralleldomain4d_wai.val.seed},
7
+ transform='${dataset.paralleldomain4d_wai.val.transform}',
8
+ data_norm_type='${dataset.paralleldomain4d_wai.val.data_norm_type}',
9
+ ROOT='${dataset.paralleldomain4d_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.paralleldomain4d_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.paralleldomain4d_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.paralleldomain4d_wai.val.variable_num_views},
13
+ num_views=${dataset.paralleldomain4d_wai.val.num_views},
14
+ covisibility_thres=${dataset.paralleldomain4d_wai.val.covisibility_thres})"
15
+ split: 'val'
16
+ dataset_resolution: ${dataset.resolution_val_paralleldomain4d}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ seed: 777
19
+ transform: 'imgnorm'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/paralleldomain4d
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.val.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/resolution_options/default.yaml ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 518_many_ar: '[(518, 518), (518, 392), (518, 336), (518, 294), (518, 252), (518, 168), (392, 518), (336, 518), (294, 518), (252, 518)]'
2
+ 518_many_landscape_ar: '[(518, 518), (518, 392), (518, 336), (518, 294), (518, 252), (518, 168)]'
3
+ 518_many_non_square_landscape_ar: '[(518, 392), (518, 336), (518, 294), (518, 252), (518, 168)]'
4
+ 518_0_50_ar: (252, 518) # 1:2
5
+ 518_0_56_ar: (294, 518) # 9:16
6
+ 518_0_66_ar: (336, 518) # 2:3
7
+ 518_0_75_ar: (392, 518) # 3:4
8
+ 518_1_00_ar: (518, 518) # 1:1
9
+ 518_1_33_ar: (518, 392) # 4:3
10
+ 518_1_52_ar: (518, 336) # 3:2
11
+ 518_1_77_ar: (518, 294) # 16:9
12
+ 518_2_00_ar: (518, 252) # 2:1
13
+ 518_3_20_ar: (518, 168) # 3.2:1
14
+ 512_many_ar: '[(512, 512), (512, 384), (512, 336), (512, 288), (512, 256), (512, 160), (384, 512), (336, 512), (288, 512), (256, 512)]'
15
+ 512_many_landscape_ar: '[(512, 512), (512, 384), (512, 336), (512, 288), (512, 256), (512, 160)]'
16
+ 512_many_non_square_landscape_ar: '[(512, 384), (512, 336), (512, 288), (512, 256), (512, 160)]'
17
+ 512_0_50_ar: (256, 512)
18
+ 512_0_56_ar: (288, 512)
19
+ 512_0_66_ar: (336, 512)
20
+ 512_0_75_ar: (384, 512)
21
+ 512_1_00_ar: (512, 512)
22
+ 512_1_33_ar: (512, 384)
23
+ 512_1_52_ar: (512, 336)
24
+ 512_1_77_ar: (512, 288)
25
+ 512_2_00_ar: (512, 256)
26
+ 512_3_20_ar: (512, 160)
27
+ 504_many_ar: '[(504, 504), (504, 378), (504, 322), (504, 280), (504, 238), (504, 154), (378, 504), (322, 504), (280, 504), (238, 504)]'
28
+ 504_many_landscape_ar: '[(504, 504), (504, 378), (504, 322), (504, 280), (504, 238), (504, 154)]'
29
+ 504_many_non_square_landscape_ar: '[(504, 378), (504, 322), (504, 280), (504, 238), (504, 154)]'
30
+ 504_0_50_ar: (238, 504)
31
+ 504_0_56_ar: (280, 504)
32
+ 504_0_66_ar: (322, 504)
33
+ 504_0_75_ar: (378, 504)
34
+ 504_1_00_ar: (504, 504)
35
+ 504_1_33_ar: (504, 378)
36
+ 504_1_52_ar: (504, 322)
37
+ 504_1_77_ar: (504, 280)
38
+ 504_2_00_ar: (504, 238)
39
+ 504_3_20_ar: (504, 154)
40
+ 448_many_ar: '[(448, 448), (448, 336), (448, 294), (448, 252), (448, 224), (448, 140), (336, 448), (294, 448), (252, 448), (224, 448)]'
41
+ 448_many_landscape_ar: '[(448, 448), (448, 336), (448, 294), (448, 252), (448, 224), (448, 140)]'
42
+ 448_many_non_square_landscape_ar: '[(448, 336), (448, 294), (448, 252), (448, 224), (448, 140)]'
43
+ 448_0_50_ar: (224, 448)
44
+ 448_0_56_ar: (252, 448)
45
+ 448_0_66_ar: (294, 448)
46
+ 448_0_75_ar: (336, 448)
47
+ 448_1_00_ar: (448, 448)
48
+ 448_1_33_ar: (448, 336)
49
+ 448_1_52_ar: (448, 294)
50
+ 448_1_77_ar: (448, 252)
51
+ 448_2_00_ar: (448, 224)
52
+ 448_3_20_ar: (448, 140)
53
+ 224_many_ar_14ps: '[(224, 224), (224, 168), (224, 154), (224, 126), (224, 112), (224, 70), (168, 224), (154, 224), (126, 224), (112, 224)]'
54
+ 224_many_landscape_ar_14ps: '[(224, 224), (224, 168), (224, 154), (224, 126), (224, 112), (224, 70)]'
55
+ 224_many_non_square_landscape_ar_14ps: '[(224, 168), (224, 154), (224, 126), (224, 112), (224, 70)]'
56
+ 224_0_50_ar_14ps: (112, 224)
57
+ 224_0_56_ar_14ps: (126, 224)
58
+ 224_0_66_ar_14ps: (154, 224)
59
+ 224_0_75_ar_14ps: (168, 224)
60
+ 224_1_00_ar: (224, 224)
61
+ 224_1_33_ar_14ps: (224, 168)
62
+ 224_1_52_ar_14ps: (224, 154)
63
+ 224_1_77_ar_14ps: (224, 126)
64
+ 224_2_00_ar_14ps: (224, 112)
65
+ 224_3_20_ar_14ps: (224, 70)
66
+ 224_many_ar_16ps: '[(224, 224), (224, 176), (224, 160), (224, 128), (224, 112), (224, 80), (176, 224), (160, 224), (128, 224), (112, 224)]'
67
+ 224_many_landscape_ar_16ps: '[(224, 224), (224, 176), (224, 160), (224, 128), (224, 112), (224, 80)]'
68
+ 224_many_non_square_landscape_ar_16ps: '[(224, 176), (224, 160), (224, 128), (224, 112), (224, 80)]'
69
+ 224_0_50_ar_16ps: (112, 224)
70
+ 224_0_56_ar_16ps: (128, 224)
71
+ 224_0_66_ar_16ps: (160, 224)
72
+ 224_0_75_ar_16ps: (176, 224)
73
+ 224_1_33_ar_16ps: (224, 176)
74
+ 224_1_52_ar_16ps: (224, 160)
75
+ 224_1_77_ar_16ps: (224, 128)
76
+ 224_2_00_ar_16ps: (224, 112)
77
+ 224_3_20_ar_16ps: (224, 80)
configs/dataset/sailvos3d_wai/default.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
configs/dataset/sailvos3d_wai/train/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "SAILVOS3DWAI(
3
+ split='${dataset.sailvos3d_wai.train.split}',
4
+ resolution=${dataset.sailvos3d_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.sailvos3d_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.sailvos3d_wai.train.aug_crop},
7
+ transform='${dataset.sailvos3d_wai.train.transform}',
8
+ data_norm_type='${dataset.sailvos3d_wai.train.data_norm_type}',
9
+ ROOT='${dataset.sailvos3d_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.sailvos3d_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.sailvos3d_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.sailvos3d_wai.train.variable_num_views},
13
+ num_views=${dataset.sailvos3d_wai.train.num_views},
14
+ covisibility_thres=${dataset.sailvos3d_wai.train.covisibility_thres})"
15
+ split: 'train'
16
+ dataset_resolution: ${dataset.resolution_train}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ aug_crop: 16
19
+ transform: 'colorjitter+grayscale+gaublur'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/sailvos3d
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.train.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/sailvos3d_wai/val/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "SAILVOS3DWAI(
3
+ split='${dataset.sailvos3d_wai.val.split}',
4
+ resolution=${dataset.sailvos3d_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.sailvos3d_wai.val.principal_point_centered},
6
+ seed=${dataset.sailvos3d_wai.val.seed},
7
+ transform='${dataset.sailvos3d_wai.val.transform}',
8
+ data_norm_type='${dataset.sailvos3d_wai.val.data_norm_type}',
9
+ ROOT='${dataset.sailvos3d_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.sailvos3d_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.sailvos3d_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.sailvos3d_wai.val.variable_num_views},
13
+ num_views=${dataset.sailvos3d_wai.val.num_views},
14
+ covisibility_thres=${dataset.sailvos3d_wai.val.covisibility_thres})"
15
+ split: 'val'
16
+ dataset_resolution: ${dataset.resolution_val_sailvos3d}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ seed: 777
19
+ transform: 'imgnorm'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/sailvos3d
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.val.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/scannetpp_wai/default.yaml ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
4
+ - test: default
configs/dataset/scannetpp_wai/test/default.yaml ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "ScanNetPPWAI(
3
+ split='${dataset.scannetpp_wai.test.split}',
4
+ resolution=${dataset.scannetpp_wai.test.dataset_resolution},
5
+ principal_point_centered=${dataset.scannetpp_wai.test.principal_point_centered},
6
+ seed=${dataset.scannetpp_wai.test.seed},
7
+ transform='${dataset.scannetpp_wai.test.transform}',
8
+ data_norm_type='${dataset.scannetpp_wai.test.data_norm_type}',
9
+ ROOT='${dataset.scannetpp_wai.test.ROOT}',
10
+ dataset_metadata_dir='${dataset.scannetpp_wai.test.dataset_metadata_dir}',
11
+ variable_num_views=${dataset.scannetpp_wai.test.variable_num_views},
12
+ num_views=${dataset.scannetpp_wai.test.num_views},
13
+ covisibility_thres=${dataset.scannetpp_wai.test.covisibility_thres})"
14
+ split: 'test'
15
+ dataset_resolution: ${dataset.resolution_test_scannetpp}
16
+ principal_point_centered: ${dataset.principal_point_centered}
17
+ seed: 777
18
+ transform: 'imgnorm'
19
+ data_norm_type: ${model.data_norm_type}
20
+ ROOT: ${root_data_dir}/scannetppv2
21
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
22
+ variable_num_views: ${dataset.test.variable_num_views}
23
+ num_views: ${dataset.num_views}
24
+ covisibility_thres: 0.25
configs/dataset/scannetpp_wai/train/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "ScanNetPPWAI(
3
+ split='${dataset.scannetpp_wai.train.split}',
4
+ resolution=${dataset.scannetpp_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.scannetpp_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.scannetpp_wai.train.aug_crop},
7
+ transform='${dataset.scannetpp_wai.train.transform}',
8
+ data_norm_type='${dataset.scannetpp_wai.train.data_norm_type}',
9
+ ROOT='${dataset.scannetpp_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.scannetpp_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.scannetpp_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.scannetpp_wai.train.variable_num_views},
13
+ num_views=${dataset.scannetpp_wai.train.num_views},
14
+ covisibility_thres=${dataset.scannetpp_wai.train.covisibility_thres})"
15
+ split: 'train'
16
+ dataset_resolution: ${dataset.resolution_train}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ aug_crop: 16
19
+ transform: 'colorjitter+grayscale+gaublur'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/scannetppv2
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.train.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/scannetpp_wai/val/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "ScanNetPPWAI(
3
+ split='${dataset.scannetpp_wai.val.split}',
4
+ resolution=${dataset.scannetpp_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.scannetpp_wai.val.principal_point_centered},
6
+ seed=${dataset.scannetpp_wai.val.seed},
7
+ transform='${dataset.scannetpp_wai.val.transform}',
8
+ data_norm_type='${dataset.scannetpp_wai.val.data_norm_type}',
9
+ ROOT='${dataset.scannetpp_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.scannetpp_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.scannetpp_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.scannetpp_wai.val.variable_num_views},
13
+ num_views=${dataset.scannetpp_wai.val.num_views},
14
+ covisibility_thres=${dataset.scannetpp_wai.val.covisibility_thres})"
15
+ split: 'val'
16
+ dataset_resolution: ${dataset.resolution_val_scannetpp}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ seed: 777
19
+ transform: 'imgnorm'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/scannetppv2
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.val.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/spring_wai/default.yaml ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ defaults:
2
+ - train: default
3
+ - val: default
configs/dataset/spring_wai/train/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "SpringWAI(
3
+ split='${dataset.spring_wai.train.split}',
4
+ resolution=${dataset.spring_wai.train.dataset_resolution},
5
+ principal_point_centered=${dataset.spring_wai.train.principal_point_centered},
6
+ aug_crop=${dataset.spring_wai.train.aug_crop},
7
+ transform='${dataset.spring_wai.train.transform}',
8
+ data_norm_type='${dataset.spring_wai.train.data_norm_type}',
9
+ ROOT='${dataset.spring_wai.train.ROOT}',
10
+ dataset_metadata_dir='${dataset.spring_wai.train.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.spring_wai.train.overfit_num_sets},
12
+ variable_num_views=${dataset.spring_wai.train.variable_num_views},
13
+ num_views=${dataset.spring_wai.train.num_views},
14
+ covisibility_thres=${dataset.spring_wai.train.covisibility_thres})"
15
+ split: 'train'
16
+ dataset_resolution: ${dataset.resolution_train}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ aug_crop: 16
19
+ transform: 'colorjitter+grayscale+gaublur'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/spring
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.train.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25
configs/dataset/spring_wai/val/default.yaml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset_str:
2
+ "SpringWAI(
3
+ split='${dataset.spring_wai.val.split}',
4
+ resolution=${dataset.spring_wai.val.dataset_resolution},
5
+ principal_point_centered=${dataset.spring_wai.val.principal_point_centered},
6
+ seed=${dataset.spring_wai.val.seed},
7
+ transform='${dataset.spring_wai.val.transform}',
8
+ data_norm_type='${dataset.spring_wai.val.data_norm_type}',
9
+ ROOT='${dataset.spring_wai.val.ROOT}',
10
+ dataset_metadata_dir='${dataset.spring_wai.val.dataset_metadata_dir}',
11
+ overfit_num_sets=${dataset.spring_wai.val.overfit_num_sets},
12
+ variable_num_views=${dataset.spring_wai.val.variable_num_views},
13
+ num_views=${dataset.spring_wai.val.num_views},
14
+ covisibility_thres=${dataset.spring_wai.val.covisibility_thres})"
15
+ split: 'val'
16
+ dataset_resolution: ${dataset.resolution_val_spring}
17
+ principal_point_centered: ${dataset.principal_point_centered}
18
+ seed: 777
19
+ transform: 'imgnorm'
20
+ data_norm_type: ${model.data_norm_type}
21
+ ROOT: ${root_data_dir}/spring
22
+ dataset_metadata_dir: ${mapanything_dataset_metadata_dir}
23
+ overfit_num_sets: null
24
+ variable_num_views: ${dataset.val.variable_num_views}
25
+ num_views: ${dataset.num_views}
26
+ covisibility_thres: 0.25