data_source.yaml 1.76 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
45
46
47
48
49
50
51
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
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
52
53
54
55
56
57
58
59
60
61
data_loader_map_provider_SequenceDataLoaderMapProvider_args:
  batch_size: 1
  num_workers: 0
  dataset_len: 1000
  dataset_len_val: 1
  images_per_seq_options:
  - 2
  sample_consecutive_frames: false
  consecutive_frames_max_gap: 0
  consecutive_frames_max_gap_seconds: 0.1