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/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,