423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-16 20:30:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

pluggable JsonIndexDataset

Browse Source 
Summary: Make dataset type and args configurable on JsonIndexDatasetMapProvider.

Reviewed By: davnov134

Differential Revision: D36666705

fbshipit-source-id: 4d0a3781d9a956504f51f1c7134c04edf1eb2846
This commit is contained in:
Jeremy Reizenstein
2022-06-10 12:22:46 -07:00
committed by Facebook GitHub Bot
parent 1d43251391
commit 6275283202
8 changed files with 117 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
projects/implicitron_trainer/configs/repro_base.yaml
View File

@@ -24,12 +24,13 @@ data_source_args:
- 10
dataset_map_provider_JsonIndexDatasetMapProvider_args:
dataset_root: ${oc.env:CO3D_DATASET_ROOT}
load_point_clouds: false
mask_depths: false
mask_images: false
n_frames_per_sequence: -1
test_on_train: true
test_restrict_sequence_id: 0
dataset_JsonIndexDataset_args:
load_point_clouds: false
mask_depths: false
mask_images: false
generic_model_args:
loss_weights:
loss_mask_bce: 1.0

3
projects/implicitron_trainer/configs/repro_multiseq_base.yaml
View File

@@ -20,9 +20,6 @@ data_source_args:
dataset_map_provider_JsonIndexDatasetMapProvider_args:
assert_single_seq: false
task_str: multisequence
load_point_clouds: false
mask_depths: false
mask_images: false
n_frames_per_sequence: -1
test_on_train: true
test_restrict_sequence_id: 0