From 1bdab5ecb6ec99d0cfb595fc78d03ad4a3d5dae0 Mon Sep 17 00:00:00 2001 From: terrance Date: Fri, 12 May 2023 19:07:03 +0000 Subject: [PATCH] undo chagnes in data folder --- .../data/datamanagers/sdf_datamanager.py | 48 ------------------- .../dataparsers/arkitscenes_dataparser.py | 1 - .../data/dataparsers/scannet_dataparser.py | 1 - 3 files changed, 50 deletions(-) delete mode 100644 nerfstudio/data/datamanagers/sdf_datamanager.py diff --git a/nerfstudio/data/datamanagers/sdf_datamanager.py b/nerfstudio/data/datamanagers/sdf_datamanager.py deleted file mode 100644 index e1aca46df..000000000 --- a/nerfstudio/data/datamanagers/sdf_datamanager.py +++ /dev/null @@ -1,48 +0,0 @@ -# Copyright 2022 The Nerfstudio Team. All rights reserved. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - -""" -SDF datamanager. -""" - -from dataclasses import dataclass, field -from typing import Type - -from nerfstudio.data.datamanagers import base_datamanager -from nerfstudio.data.datasets.sdf_dataset import SDFDataset - - -@dataclass -class SDFDataManagerConfig(base_datamanager.VanillaDataManagerConfig): - """A sdf datamanager - required to use with .setup()""" - - _target: Type = field(default_factory=lambda: SDFDataManager) - - -class SDFDataManager(base_datamanager.VanillaDataManager): # pylint: disable=abstract-method - """Data manager implementation for data that also requires processing sdf data. - Args: - config: the DataManagerConfig used to instantiate class - """ - - def create_train_dataset(self) -> SDFDataset: - self.train_dataparser_outputs = self.dataparser.get_dataparser_outputs(split="train") - return SDFDataset( - dataparser_outputs=self.train_dataparser_outputs, - ) - - def create_eval_dataset(self) -> SDFDataset: - return SDFDataset( - dataparser_outputs=self.dataparser.get_dataparser_outputs(split=self.test_split), - ) diff --git a/nerfstudio/data/dataparsers/arkitscenes_dataparser.py b/nerfstudio/data/dataparsers/arkitscenes_dataparser.py index cda832ef8..0ee105196 100644 --- a/nerfstudio/data/dataparsers/arkitscenes_dataparser.py +++ b/nerfstudio/data/dataparsers/arkitscenes_dataparser.py @@ -21,7 +21,6 @@ from typing import Literal, Type import cv2 import numpy as np import torch -from typing_extensions import Literal from nerfstudio.cameras import camera_utils from nerfstudio.cameras.cameras import Cameras, CameraType diff --git a/nerfstudio/data/dataparsers/scannet_dataparser.py b/nerfstudio/data/dataparsers/scannet_dataparser.py index eb0493926..fb863073e 100644 --- a/nerfstudio/data/dataparsers/scannet_dataparser.py +++ b/nerfstudio/data/dataparsers/scannet_dataparser.py @@ -21,7 +21,6 @@ from typing import Literal, Type import cv2 import numpy as np import torch -from typing_extensions import Literal from nerfstudio.cameras import camera_utils from nerfstudio.cameras.cameras import Cameras, CameraType