| # YOLO Custom Model | |
| This is a YOLO model trained on custom data. | |
| ## Model Description | |
| - Model Type: YOLO | |
| - Training Data: Custom Dataset | |
| - Input: Images | |
| - Output: Bounding boxes with class predictions | |
| ## Usage | |
| ```python | |
| from ultralytics import YOLO | |
| # Load the model | |
| model = YOLO('model.pt') | |
| # Make predictions | |
| results = model('image.jpg') | |
| ``` | |
| ## Training Details | |
| - Framework: Ultralytics YOLOv8 | |
| - Training Device: cuda | |