423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-14 19:36:23 +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:
"""

4
projects/implicitron_trainer/visualize_reconstruction.py
View File

@@ -25,7 +25,7 @@ from experiment import init_model
from omegaconf import OmegaConf
from pytorch3d.implicitron.dataset.data_source import ImplicitronDataSource
from pytorch3d.implicitron.dataset.dataset_base import FrameData
from pytorch3d.implicitron.dataset.implicitron_dataset import ImplicitronDataset
from pytorch3d.implicitron.dataset.json_index_dataset import JsonIndexDataset
from pytorch3d.implicitron.dataset.utils import is_train_frame
from pytorch3d.implicitron.models.base_model import EvaluationMode
from pytorch3d.implicitron.tools.configurable import get_default_args
@@ -41,7 +41,7 @@ from tqdm import tqdm
def render_sequence(
dataset: ImplicitronDataset,
dataset: JsonIndexDataset,
sequence_name: str,
model: torch.nn.Module,
video_path,