initial vision test, pc

CMU-Robotics-Club · Dec 31, 2024 · 76e2b10 · 76e2b10
1 parent 5f16f84
commit 76e2b10
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -6,4 +6,5 @@ docker-compose.yml~
 *TEMP_DO_NOT_EDIT.txt
 rb_ws/src/buggy/bags/*                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                   
 *.bag
-vision/data/*
+vision/data/*
+.vision/*
diff --git a/vision-requirements.txt b/vision-requirements.txt
@@ -0,0 +1 @@
+ultralytics
diff --git a/vision/ultralytics-train.py b/vision/ultralytics-train.py
@@ -0,0 +1,8 @@
+from ultralytics import YOLO
+
+# Load a model
+model = YOLO("yolo11n.pt")  # load a pretrained model (recommended for training)
+
+# 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")
+print(results)
diff --git a/vision/yolo11n.pt b/vision/yolo11n.pt