code: def __init__(
      self,
      image_dir: PurePath,
      basenet_preprocess: Callable[[np.array], torch.tensor],
      recursive: Optional[bool],
  ) -> None:
      self.image_dir = image_dir
      self.basenet_preprocess = basenet_preprocess
      self.recursive = recursive
      self.image_files = sorted(
          generate_files(self.image_dir, self.recursive)
      )
  
file_path: imagededup/imagededup/utils/data_generator.py
imports: from pathlib import PurePath
  from typing import Dict, Callable, Optional, List, Tuple
  import numpy
  import torch
  from torch.utils.data import Dataset, DataLoader
  from imagededup.utils.image_utils import load_image
  from imagededup.utils.general_utils import generate_files
label_desc: Use image node_id 1 to create a new ImgDataset object from inherited base classes: Dataset with example: obj = ImgDataset(image_dir, basenet_preprocess, recursive)