添加状态分类和液面分割

2025-09-01 14:14:18 +08:00
parent 6e553f6a20
commit ad52ab9125
2379 changed files with 102501 additions and 1465 deletions
--- a/ultralytics_yolo11-main/train_obb_main.py
+++ b/ultralytics_yolo11-main/train_obb_main.py
@ -4,8 +4,8 @@ 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,
+        epochs=100,
+        imgsz=640,
        batch=4,
        workers=10,
        device='0',