diff --git a/dataset.py b/dataset.py
index fed90cb3b9dca489bfbefb3011e56b5d81f32e11..447a16a633e1c661e0b3a5b14b9bca95989a0820 100644
--- a/dataset.py
+++ b/dataset.py
@@ -17,8 +17,8 @@ class GetDatasetGenerator:
             line = f.readline()  # 读取下一行
 
     def __getitem__(self, index):
-        size1 = 2048
-        size2 = 2048
+        size1 = 1536
+        size2 = 1536
         r1 = random.randint(0, 6800 - size1)
         r2 = random.randint(0, 7200 - size2)
         image = np.float32(cv2.imread(self.path+"/JPEGImages/"+self.__data[index]+".bmp"))