修改训练参数

dataset.py

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

main.py

@@ -29,7 +29,7 @@ dataset_path = '/home/GXkaifa1/hexiangdong'
 
 train_dataset_generator = GetDatasetGenerator(dataset_path + '/datasets', 'train')
 train_dataset = ds.GeneratorDataset(train_dataset_generator, ["data", "label"], shuffle=True)
-train_dataset = train_dataset.batch(2, drop_remainder=True)
+train_dataset = train_dataset.batch(4, drop_remainder=True)
 
 # lr_iter = exponential_lr(3e-5, 20, 0.9, 100, staircase=True)