bushu

ultralytics_yolo11-main/train_seg_resize_main.py

@@ -7,8 +7,8 @@ if __name__ == '__main__':
     # 开始训练
     results = model.train(
         data='/home/hx/yolo/ultralytics_yolo11-main/resize_seg_data.yaml',           # 数据配置文件
-        epochs=1000,                 # 训练轮数
-        imgsz=640,
+        epochs=100,               # 训练轮数
+        imgsz=1280,
         batch=4,                   # 每批图像数量
         workers=10,                 # 数据加载线程数
         device='0',                 # 使用 GPU 0
@@ -16,6 +16,6 @@ if __name__ == '__main__':
         name='exp',                 # 实验名称
         exist_ok=False,             # 不覆盖已有实验
         optimizer='AdamW',          # 可选优化器
-        lr0=0.001,                  # 初始学习率
-        patience=500,                # 早停轮数
+        lr0=0.0005,                  # 初始学习率
+        patience=0,                # 早停轮数
     )