From a61b88db0688815c84b380f5ca420e540136d568 Mon Sep 17 00:00:00 2001
From: luolian666 <154983879+luolian666@users.noreply.github.com>
Date: Tue, 19 Mar 2024 22:18:17 +0800
Subject: [PATCH] Update pretrain_product_dataset.py

jpg
---
 data/pretrain_product_dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/data/pretrain_product_dataset.py b/data/pretrain_product_dataset.py
index 2fcfce9..338ee09 100644
--- a/data/pretrain_product_dataset.py
+++ b/data/pretrain_product_dataset.py
@@ -43,7 +43,7 @@ def __len__(self):
     def __getitem__(self, index):
 
         item_id, caption, cate_name = self.data_list[index]
-        image_path = "{}/{}.png".format(self.image_path, item_id)
+        image_path = "{}/{}.jpg".format(self.image_path, item_id)
         try:
             image = Image.open(image_path).convert('RGB')
         except: