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