data_source.yaml 3.68 KB
Newer Older
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
1
dataset_map_provider_class_type: ???
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
2
data_loader_map_provider_class_type: SequenceDataLoaderMapProvider
3
4
5
6
7
8
9
dataset_map_provider_BlenderDatasetMapProvider_args:
  base_dir: ???
  object_name: ???
  path_manager_factory_class_type: PathManagerFactory
  n_known_frames_for_test: null
  path_manager_factory_PathManagerFactory_args:
    silence_logs: true
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
10
11
12
13
14
15
16
17
18
19
dataset_map_provider_JsonIndexDatasetMapProvider_args:
  category: ???
  task_str: singlesequence
  dataset_root: ''
  n_frames_per_sequence: -1
  test_on_train: false
  restrict_sequence_name: []
  test_restrict_sequence_id: -1
  assert_single_seq: false
  only_test_set: false
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
  dataset_class_type: JsonIndexDataset
  path_manager_factory_class_type: PathManagerFactory
  dataset_JsonIndexDataset_args:
    limit_to: 0
    limit_sequences_to: 0
    exclude_sequence: []
    limit_category_to: []
    load_images: true
    load_depths: true
    load_depth_masks: true
    load_masks: true
    load_point_clouds: false
    max_points: 0
    mask_images: false
    mask_depths: false
    image_height: 800
    image_width: 800
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
37
    box_crop: true
38
    box_crop_mask_thr: 0.4
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
39
40
    box_crop_context: 0.3
    remove_empty_masks: true
41
42
    seed: 0
    sort_frames: false
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
43
44
  path_manager_factory_PathManagerFactory_args:
    silence_logs: true
David Novotny's avatar
David Novotny committed
45
46
47
48
49
50
51
dataset_map_provider_JsonIndexDatasetMapProviderV2_args:
  category: ???
  subset_name: ???
  dataset_root: ''
  test_on_train: false
  only_test_set: false
  load_eval_batches: true
52
  num_load_workers: 4
53
  n_known_frames_for_test: 0
David Novotny's avatar
David Novotny committed
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
  dataset_class_type: JsonIndexDataset
  path_manager_factory_class_type: PathManagerFactory
  dataset_JsonIndexDataset_args:
    limit_to: 0
    limit_sequences_to: 0
    pick_sequence: []
    exclude_sequence: []
    limit_category_to: []
    load_images: true
    load_depths: true
    load_depth_masks: true
    load_masks: true
    load_point_clouds: false
    max_points: 0
    mask_images: false
    mask_depths: false
    image_height: 800
    image_width: 800
    box_crop: true
    box_crop_mask_thr: 0.4
    box_crop_context: 0.3
    remove_empty_masks: true
    n_frames_per_sequence: -1
    seed: 0
    sort_frames: false
  path_manager_factory_PathManagerFactory_args:
    silence_logs: true
81
82
83
84
85
86
87
dataset_map_provider_LlffDatasetMapProvider_args:
  base_dir: ???
  object_name: ???
  path_manager_factory_class_type: PathManagerFactory
  n_known_frames_for_test: null
  path_manager_factory_PathManagerFactory_args:
    silence_logs: true
88
  downscale_factor: 4
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
89
90
91
92
dataset_map_provider_RenderedMeshDatasetMapProvider_args:
  num_views: 40
  data_file: null
  azimuth_range: 180.0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
93
  distance: 2.7
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
94
95
  resolution: 128
  use_point_light: true
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
96
  gpu_idx: 0
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
97
98
99
  path_manager_factory_class_type: PathManagerFactory
  path_manager_factory_PathManagerFactory_args:
    silence_logs: true
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
100
101
102
data_loader_map_provider_SequenceDataLoaderMapProvider_args:
  batch_size: 1
  num_workers: 0
103
104
105
106
107
108
109
  dataset_length_train: 0
  dataset_length_val: 0
  dataset_length_test: 0
  train_conditioning_type: SAME
  val_conditioning_type: SAME
  test_conditioning_type: KNOWN
  images_per_seq_options: []
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
110
111
112
  sample_consecutive_frames: false
  consecutive_frames_max_gap: 0
  consecutive_frames_max_gap_seconds: 0.1
113
114
115
116
117
118
data_loader_map_provider_SimpleDataLoaderMapProvider_args:
  batch_size: 1
  num_workers: 0
  dataset_length_train: 0
  dataset_length_val: 0
  dataset_length_test: 0
119
120
121
122
123
124
125
126
127
128
129
130
131
data_loader_map_provider_TrainEvalDataLoaderMapProvider_args:
  batch_size: 1
  num_workers: 0
  dataset_length_train: 0
  dataset_length_val: 0
  dataset_length_test: 0
  train_conditioning_type: SAME
  val_conditioning_type: SAME
  test_conditioning_type: KNOWN
  images_per_seq_options: []
  sample_consecutive_frames: false
  consecutive_frames_max_gap: 0
  consecutive_frames_max_gap_seconds: 0.1