This commit is contained in:
2025-12-19 06:37:19 +01:00
parent 58e19bca64
commit 9c217c0599
29 changed files with 346 additions and 2 deletions

18
rpi/training/training.py Normal file
View File

@@ -0,0 +1,18 @@
from ultralytics import YOLO
def main():
model = YOLO("models/yolo11n.pt")
model.train(
data="./datasets/unified/data.yaml",
epochs=200,
patience=30,
imgsz=640,
batch=12,
device=0,
project="result",
name="unified-training",
exist_ok=True
)
if __name__ == "__main__":
main()