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])