diff --git a/.gitignore b/.gitignore index 61bda68..6c881e8 100644 --- a/.gitignore +++ b/.gitignore @@ -6,4 +6,5 @@ docker-compose.yml~ *TEMP_DO_NOT_EDIT.txt rb_ws/src/buggy/bags/* *.bag -vision/data/* \ No newline at end of file +vision/data/* +.vision/* \ No newline at end of file diff --git a/vision-requirements.txt b/vision-requirements.txt new file mode 100644 index 0000000..c6856f7 --- /dev/null +++ b/vision-requirements.txt @@ -0,0 +1 @@ +ultralytics \ No newline at end of file diff --git a/vision/ultralytics-train.py b/vision/ultralytics-train.py new file mode 100644 index 0000000..9e67c6a --- /dev/null +++ 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) \ No newline at end of file diff --git a/vision/yolo11n.pt b/vision/yolo11n.pt new file mode 100644 index 0000000..45b273b Binary files /dev/null and b/vision/yolo11n.pt differ