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

8
tests/implicitron/test_dataset_visualize.py
View File

@@ -12,7 +12,7 @@ import unittest
import torch
import torchvision
from pytorch3d.implicitron.dataset.implicitron_dataset import ImplicitronDataset
from pytorch3d.implicitron.dataset.json_index_dataset import JsonIndexDataset
from pytorch3d.implicitron.dataset.visualize import get_implicitron_sequence_pointcloud
from pytorch3d.implicitron.tools.point_cloud_utils import render_point_cloud_pytorch3d
from pytorch3d.vis.plotly_vis import plot_scene
@@ -38,7 +38,7 @@ class TestDatasetVisualize(unittest.TestCase):
sequence_file = os.path.join(dataset_root, category, "sequence_annotations.jgz")
self.image_size = 256
self.datasets = {
"simple": ImplicitronDataset(
"simple": JsonIndexDataset(
frame_annotations_file=frame_file,
sequence_annotations_file=sequence_file,
dataset_root=dataset_root,
@@ -48,7 +48,7 @@ class TestDatasetVisualize(unittest.TestCase):
load_point_clouds=True,
path_manager=path_manager,
),
"nonsquare": ImplicitronDataset(
"nonsquare": JsonIndexDataset(
frame_annotations_file=frame_file,
sequence_annotations_file=sequence_file,
dataset_root=dataset_root,
@@ -58,7 +58,7 @@ class TestDatasetVisualize(unittest.TestCase):
load_point_clouds=True,
path_manager=path_manager,
),
"nocrop": ImplicitronDataset(
"nocrop": JsonIndexDataset(
frame_annotations_file=frame_file,
sequence_annotations_file=sequence_file,
dataset_root=dataset_root,

4
tests/implicitron/test_evaluation.py
View File

@@ -15,7 +15,7 @@ import unittest
import lpips
import torch
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.evaluation.evaluate_new_view_synthesis import eval_batch
from pytorch3d.implicitron.models.base_model import ImplicitronModelBase
from pytorch3d.implicitron.models.generic_model import GenericModel # noqa
@@ -44,7 +44,7 @@ class TestEvaluation(unittest.TestCase):
frame_file = os.path.join(dataset_root, category, "frame_annotations.jgz")
sequence_file = os.path.join(dataset_root, category, "sequence_annotations.jgz")
self.image_size = 64
self.dataset = ImplicitronDataset(
self.dataset = JsonIndexDataset(
frame_annotations_file=frame_file,
sequence_annotations_file=sequence_file,
dataset_root=dataset_root,