bryandts commited on
Commit
8072e11
1 Parent(s): a305b79

Update preprocessing.py

Browse files
Files changed (1) hide show
  1. preprocessing.py +1 -1
preprocessing.py CHANGED
@@ -15,7 +15,7 @@ class Text2ImageDataset(Dataset):
15
  self.dataset_dir = dataset_dir
16
  with open(os.path.join(self.dataset_dir, 'descriptions.json'), 'r') as file:
17
  self.dataset = json.load(file)
18
- self.images_path = os.path.join(dataset_dir, 'CUHKSZ_Photos')
19
 
20
  def __len__(self):
21
  return len(self.dataset)
 
15
  self.dataset_dir = dataset_dir
16
  with open(os.path.join(self.dataset_dir, 'descriptions.json'), 'r') as file:
17
  self.dataset = json.load(file)
18
+ self.images_path = os.path.join(dataset_dir)
19
 
20
  def __len__(self):
21
  return len(self.dataset)