423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2026-02-27 00:36:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

rename ImplicitronDataset to JsonIndexDataset

Browse Source 
Summary: The ImplicitronDataset class corresponds to JsonIndexDatasetMapProvider

Reviewed By: shapovalov

Differential Revision: D36661396

fbshipit-source-id: 80ca2ff81ef9ecc2e3d1f4e1cd14b6f66a7ec34d
This commit is contained in:
Jeremy Reizenstein
2022-05-25 10:16:59 -07:00
committed by Facebook GitHub Bot
parent 34f648ede0
commit fbd3c679ac
8 changed files with 25 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
projects/implicitron_trainer/experiment.py
View File

@@ -68,7 +68,7 @@ from pytorch3d.implicitron.dataset.data_loader_map_provider import DataLoaderMap
from pytorch3d.implicitron.dataset.data_source import ImplicitronDataSource, Task
from pytorch3d.implicitron.dataset.dataset_base import FrameData
from pytorch3d.implicitron.dataset.dataset_map_provider import DatasetMap
from pytorch3d.implicitron.dataset.implicitron_dataset import ImplicitronDataset
from pytorch3d.implicitron.dataset.json_index_dataset import JsonIndexDataset
from pytorch3d.implicitron.evaluation import evaluate_new_view_synthesis as evaluate
from pytorch3d.implicitron.models.generic_model import EvaluationMode, GenericModel
from pytorch3d.implicitron.tools import model_io, vis_utils
@@ -641,7 +641,7 @@ def _run_eval(model, all_source_cameras, loader, task: Task, device):
def _get_all_source_cameras(
dataset: ImplicitronDataset,
dataset: JsonIndexDataset,
num_workers: int = 8,
) -> CamerasBase:
"""