bushu

2025-10-21 14:11:52 +08:00
parent 349449f2b7
commit df7c0730f5
363 changed files with 5386 additions and 578 deletions
--- a/ultralytics_yolo11-main/train_ailai_main.py
+++ b/ultralytics_yolo11-main/train_ailai_main.py
@ -0,0 +1,19 @@
+from ultralytics import YOLO
+
+if __name__ == '__main__':
+    #model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/ultralytics/cfg/models/11/yolo11-obb.yaml')
+    model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai3/weights/last.pt')
+    results = model.train(
+        data='data_ailai.yaml',
+        epochs=300,
+        imgsz=640,
+        batch=4,
+        workers=10,
+        device='0',
+        project='runs/train',
+        name='exp_ailai',
+        exist_ok=False,
+        optimizer='AdamW',
+        lr0=0.0001,
+        patience=0,
+    )