data_source.yaml 1.28 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
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
3
4
5
6
7
8
9
10
11
12
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
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
  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
30
    box_crop: true
31
    box_crop_mask_thr: 0.4
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
32
33
    box_crop_context: 0.3
    remove_empty_masks: true
34
35
    seed: 0
    sort_frames: false
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
36
37
  path_manager_factory_PathManagerFactory_args:
    silence_logs: true
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
38
39
40
41
42
43
44
45
46
47
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