From 4520f8366eaf7bcc84ea505374fb0d5d879035c0 Mon Sep 17 00:00:00 2001 From: Bingxin Date: Wed, 10 Dec 2025 10:00:06 +0100 Subject: [PATCH] [FIX] use the same DatasetMode from depth base_depth_dataset (alternative of #167) --- src/dataset/base_iid_dataset.py | 7 +------ src/dataset/base_normals_dataset.py | 7 +------ 2 files changed, 2 insertions(+), 12 deletions(-) diff --git a/src/dataset/base_iid_dataset.py b/src/dataset/base_iid_dataset.py index a7c302f0..5748cb69 100644 --- a/src/dataset/base_iid_dataset.py +++ b/src/dataset/base_iid_dataset.py @@ -41,6 +41,7 @@ from enum import Enum from torch.utils.data import Dataset from torchvision.transforms import InterpolationMode, Resize +from .base_depth_dataset import DatasetMode from src.util.image_util import ( img_hwc2chw, @@ -51,12 +52,6 @@ ) -class DatasetMode(Enum): - RGB_ONLY = "rgb_only" - EVAL = "evaluate" - TRAIN = "train" - - class BaseIIDDataset(Dataset): def __init__( self, diff --git a/src/dataset/base_normals_dataset.py b/src/dataset/base_normals_dataset.py index 6b5cc361..e90408c6 100644 --- a/src/dataset/base_normals_dataset.py +++ b/src/dataset/base_normals_dataset.py @@ -39,12 +39,7 @@ import torchvision.transforms.functional as TF from torch.utils.data import Dataset, get_worker_info from torchvision.transforms import InterpolationMode, Resize, ColorJitter - - -class DatasetMode(Enum): - RGB_ONLY = "rgb_only" - EVAL = "evaluate" - TRAIN = "train" +from .base_depth_dataset import DatasetMode class BaseNormalsDataset(Dataset):