增加obb

2025-08-18 15:13:54 +08:00
parent 254afbbc43
commit 6e553f6a20
700 changed files with 576 additions and 1390 deletions
--- a/ultralytics_yolo11-main/train_obb_main.py
+++ b/ultralytics_yolo11-main/train_obb_main.py
@ -0,0 +1,18 @@
+from ultralytics import YOLO
+
+if __name__ == '__main__':
+    model = YOLO(r'/home/hx/yolo/ultralytics_yolo11-main/ultralytics/cfg/models/11/yolo11-obb.yaml')
+    results = model.train(
+        data='obb_data.yaml',
+        epochs=1000,
+        imgsz=1280,
+        batch=4,
+        workers=10,
+        device='0',
+        project='runs/train',
+        name='exp_obb',
+        exist_ok=False,
+        optimizer='AdamW',
+        lr0=0.001,
+        patience=20,
+    )