Handoff completed training into closed loop

scripts/run_belgium_building_training_loop.py

@@ -101,6 +101,9 @@ def training_command(
     mosaic: float = 1.0,
     scale: float = 0.5,
     translate: float = 0.1,
+    degrees: float = 0.0,
+    flipud: float = 0.0,
+    fliplr: float = 0.5,
 ) -> list[str]:
     command = [
         yolo,
@@ -120,6 +123,9 @@ def training_command(
         f"mosaic={mosaic}",
         f"scale={scale}",
         f"translate={translate}",
+        f"degrees={degrees}",
+        f"flipud={flipud}",
+        f"fliplr={fliplr}",
         f"seed={seed}",
         "deterministic=True",
         f"project={project}",
@@ -183,6 +189,9 @@ def main() -> int:
     parser.add_argument("--mosaic", type=float, default=1.0)
     parser.add_argument("--scale", type=float, default=0.5)
     parser.add_argument("--translate", type=float, default=0.1)
+    parser.add_argument("--degrees", type=float, default=0.0)
+    parser.add_argument("--flipud", type=float, default=0.0)
+    parser.add_argument("--fliplr", type=float, default=0.5)
     parser.add_argument("--seed", type=int, default=20260731)
     parser.add_argument("--yolo", default="yolo")
     parser.add_argument("--min-aggregate-f1", type=float, default=0.55)
@@ -246,6 +255,9 @@ def main() -> int:
             mosaic=args.mosaic,
             scale=args.scale,
             translate=args.translate,
+            degrees=args.degrees,
+            flipud=args.flipud,
+            fliplr=args.fliplr,
         )
         if args.dry_run:
             print(json.dumps({"training_command": command, "evaluate_existing": evaluate_existing}, indent=2))