Zhonathon commited on
Commit
c924ab0
1 Parent(s): a4a3c00

models path4

Browse files
recommendation/datasets/img_preprocess.py CHANGED
@@ -35,7 +35,7 @@ class Image_preprocess(data.Dataset):
35
  self.img_ids = img_ids['img_id'].values
36
  # self.sentence_clip_preprocess2 = sentence_clip_preprocess2
37
  def preprocess(self, img_id):
38
- img = Image.open(r'E:\data\processed_img' + '/' + str(img_id) + '.jpg')
39
  img = sentence_clip_preprocess2(img)
40
  return img
41
 
 
35
  self.img_ids = img_ids['img_id'].values
36
  # self.sentence_clip_preprocess2 = sentence_clip_preprocess2
37
  def preprocess(self, img_id):
38
+ img = Image.open('./imgs' + '/' + str(img_id) + '.jpg')
39
  img = sentence_clip_preprocess2(img)
40
  return img
41