dataset_map_provider

Summary: replace dataset_zoo with a pluggable DatasetMapProvider. The logic is now in annotated_file_dataset_map_provider.

Reviewed By: shapovalov

Differential Revision: D36443965

fbshipit-source-id: 9087649802810055e150b2fbfcc3c197a761f28a

projects/implicitron_trainer/configs/repro_base.yaml

@@ -6,6 +6,7 @@ architecture: generic
 visualize_interval: 0
 visdom_port: 8097
 data_source_args:
+  dataset_provider_class_type: JsonIndexDatasetMapProvider
   dataloader_args:
     batch_size: 10
     dataset_len: 1000
@@ -21,7 +22,7 @@ data_source_args:
     - 8
     - 9
     - 10
-  dataset_args:
+  dataset_map_provider_JsonIndexDatasetMapProvider_args:
     dataset_root: ${oc.env:CO3D_DATASET_ROOT}
     load_point_clouds: false
     mask_depths: false

projects/implicitron_trainer/configs/repro_multiseq_base.yaml

@@ -17,9 +17,9 @@ data_source_args:
     - 8
     - 9
     - 10
-  dataset_args:
+  dataset_map_provider_JsonIndexDatasetMapProvider_args:
     assert_single_seq: false
-    dataset_name: co3d_multisequence
+    task_str: multisequence
     load_point_clouds: false
     mask_depths: false
     mask_images: false

projects/implicitron_trainer/configs/repro_singleseq_base.yaml

@@ -9,8 +9,8 @@ data_source_args:
     num_workers: 8
     images_per_seq_options:
     - 2
-  dataset_args:
-    dataset_name: co3d_singlesequence
+  dataset_map_provider_JsonIndexDatasetMapProvider_args:
+    dataset_name: singlesequence
     assert_single_seq: true
     n_frames_per_sequence: -1
     test_restrict_sequence_id: 0