From cf65cc328fac7aaed3a42c198247d335d64f559e Mon Sep 17 00:00:00 2001
From: deepr <hexiangdong2020@outlook.com>
Date: Wed, 11 Aug 2021 16:40:17 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AE=AD=E7=BB=83=E8=BF=87=E7=A8=8B=E4=B8=AD?=
=?UTF-8?q?=E8=A3=81=E5=88=87=E5=9B=BE=E5=83=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
dataset.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/dataset.py b/dataset.py
index c994e54..4263383 100644
--- a/dataset.py
+++ b/dataset.py
@@ -17,8 +17,8 @@ class GetDatasetGenerator:
line = f.readline() # 璇诲彇涓嬩竴琛�
def __getitem__(self, index):
- size1 = 3400
- size2 = 3600
+ size1 = 3000
+ size2 = 3000
r1 = random.randint(0, 6800 - size1)
r2 = random.randint(0, 7200 - size2)
image = np.float32(cv2.imread(self.path+"/JPEGImages/"+self.__data[index]+".bmp"))
--
GitLab