viciopoli commited on
Commit
c978756
·
verified ·
1 Parent(s): a6d5bca

Update configs/depth.yaml

Browse files
Files changed (1) hide show
  1. configs/depth.yaml +2 -30
configs/depth.yaml CHANGED
@@ -1,4 +1,5 @@
1
  base_checkpoint: "./checkpoints/dune_vitbase14_448_paper.pth"
 
2
  realm_checkpoint: "./checkpoints/REALM_final_rank32_0955116.pth"
3
  embedding:
4
  num_bins: 5
@@ -14,33 +15,4 @@ head:
14
  embed_dim: 768
15
  min_depth: 1.95
16
  max_depth: 82.0
17
- num_bins: 256
18
-
19
- data:
20
- batch_size: 32
21
-
22
- datasets_config:
23
- datasets:
24
- - MVSEC
25
-
26
- common:
27
- events_representation: voxel_grid
28
- nr_bins_per_data: 5
29
- normalize_event: true
30
- sequence_length: 1
31
- target_size: [346, 260]
32
-
33
- MVSEC:
34
- root: ./datasets/MVSEC
35
- n_ev: 150000
36
- ms_ev: -1
37
- stride: 1
38
- mask: true
39
- skip_init_frames: 1
40
- depth: true
41
- min_n_ev: 5000
42
- transforms:
43
- - type: resize
44
- params:
45
- target_size: [260, 346]
46
- keep_aspect_ratio: true
 
1
  base_checkpoint: "./checkpoints/dune_vitbase14_448_paper.pth"
2
+ dune_checkpoint: "./checkpoints/dune_vitbase14_448_paper.pth"
3
  realm_checkpoint: "./checkpoints/REALM_final_rank32_0955116.pth"
4
  embedding:
5
  num_bins: 5
 
15
  embed_dim: 768
16
  min_depth: 1.95
17
  max_depth: 82.0
18
+ num_bins: 256