class_docstr: Characterizes a dataset for PyTorch
docstr: Initialization
code: def __init__(self, data_folder, transform, stage):
      "Initialization"
      self.images = glob.glob("{}/{}/*".format(data_folder, stage))
      self.transform = transforms.Compose(transform)
      "Get image size"
      _, self.img_size = Image.open(self.images[0]).size
  
file_path: gans-with-pytorch/pix2pix/datasets.py
imports: from PIL import Image
  from torch.utils import data
  import glob
  import torchvision.transforms
label_desc: Use image node_id 1 to create a new Dataset object from inherited base classes: data with example: obj = Dataset(data_folder, transform, stage)