initial model

CMU-Robotics-Club · Jan 2, 2025 · edb14cb · edb14cb
1 parent 76e2b10
commit edb14cb
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -7,4 +7,5 @@ docker-compose.yml~
 rb_ws/src/buggy/bags/*                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                   
 *.bag
 vision/data/*
-.vision/*
+.vision/*
+runs/*
diff --git a/vision/best.pt b/vision/best.pt
diff --git a/vision/ultralytics-train.py b/vision/ultralytics-train.py
@@ -2,7 +2,9 @@
 
 # Load a model
 model = YOLO("yolo11n.pt")  # load a pretrained model (recommended for training)
+yaml_path = "C:\\Users\\Mehul Goel\\Documents\\robobuggy-software\\vision\\data\\Buggy Vision.v1i.yolov11\\data.yaml"
+
 
 # Train the model with MPS
-results = model.train(data="/Users/mehulgoel/Documents/robobuggy-software/vision/data/Buggy Vision.v1i.yolov11/data.yaml", epochs=100, imgsz=640, device="mps")
+results = model.train(data=yaml_path, epochs=100, imgsz=640)
 print(results)