From ed62e718d7f24444dc038c72a94b9072577c70cd Mon Sep 17 00:00:00 2001 From: vlpr Date: Wed, 15 May 2024 10:51:27 +0000 Subject: [PATCH] remove dir from path --- src/common/data.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/common/data.py b/src/common/data.py index e12c839..d2e44e2 100644 --- a/src/common/data.py +++ b/src/common/data.py @@ -9,8 +9,6 @@ from torch.utils.data import Dataset, DataLoader from torch.utils.data.distributed import DistributedSampler from pathlib import Path -sys.path.insert(0, "../") # run under the project directory - image_extensions = ['.jpg', '.png'] def load_images_cached(images_dir_path): image_paths = sorted([f for f in Path(images_dir_path).glob("*") if f.suffix.lower() in image_extensions])