ImplicitronDatasetBase -> DatasetBase

Summary: Just a rename

Reviewed By: shapovalov

Differential Revision: D36516885

fbshipit-source-id: 2126e3aee26d89a95afdb31e06942d61cbe88d5a
This commit is contained in:
Jeremy Reizenstein
2022-05-20 07:50:30 -07:00
committed by Facebook GitHub Bot
parent 0f12c51646
commit 9fe15da3cd
7 changed files with 17 additions and 17 deletions

View File

@@ -9,7 +9,7 @@ import unittest
from collections import defaultdict
from dataclasses import dataclass
from pytorch3d.implicitron.dataset.dataset_base import ImplicitronDatasetBase
from pytorch3d.implicitron.dataset.dataset_base import DatasetBase
from pytorch3d.implicitron.dataset.scene_batch_sampler import SceneBatchSampler
@@ -19,7 +19,7 @@ class MockFrameAnnotation:
frame_timestamp: float = 0.0
class MockDataset(ImplicitronDatasetBase):
class MockDataset(DatasetBase):
def __init__(self, num_seq, max_frame_gap=1):
"""
Makes a gap of max_frame_gap frame numbers in the middle of each sequence