| # scripts/download_yolov8_model.py | |
| from ultralytics import YOLO | |
| import os | |
| # Directory to save the model | |
| MODEL_DIR = "models" | |
| os.makedirs(MODEL_DIR, exist_ok=True) | |
| # Path to save the model | |
| MODEL_PATH = os.path.join(MODEL_DIR, "yolov8_model.pt") | |
| # Download pre-trained YOLOv8 nano model | |
| try: | |
| model = YOLO("yolov8n.pt") # Automatically downloads from Ultralytics | |
| model.save(MODEL_PATH) | |
| print(f"Model saved to {MODEL_PATH}") | |
| except Exception as e: | |
| print(f"Failed to download or save model: {str(e)}") |