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(
|
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 |
|