dschandra commited on
Commit
111eec1
·
verified ·
1 Parent(s): 20f24a7

Update models/yolov8_model.pt

Browse files
Files changed (1) hide show
  1. models/yolov8_model.pt +17 -0
models/yolov8_model.pt CHANGED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # scripts/download_yolov8_model.py
2
+ from ultralytics import YOLO
3
+ import os
4
+
5
+ # Directory to save the model
6
+ MODEL_DIR = "models"
7
+ os.makedirs(MODEL_DIR, exist_ok=True)
8
+
9
+ # Path to save the model
10
+ MODEL_PATH = os.path.join(MODEL_DIR, "yolov8_model.pt")
11
+
12
+ # Download pre-trained YOLOv8 nano model
13
+ model = YOLO("yolov8n.pt") # Automatically downloads from Ultralytics
14
+
15
+ # Save the model to models/yolov8_model.pt
16
+ model.save(MODEL_PATH)
17
+ print(f"Model saved to {MODEL_PATH}")