| --- |
| tags: |
| - ultralytics |
| - yolov8 |
| - object-detection |
| - rock-classification |
| license: apache-2.0 |
| datasets: |
| - your-dataset-name |
| model-index: |
| - name: Rock vs Non-Rock YOLOv8 Model |
| results: |
| - task: |
| type: object-detection |
| dataset: |
| name: Rock vs Non-Rock Dataset |
| type: custom |
| metrics: |
| - type: mAP |
| value: 0.85 |
| --- |
| |
| # Rock vs Non-Rock YOLOv8 Model |
|
|
| This model was trained using YOLOv8 to classify rocks vs. non-rocks in an industrial setting. It is designed to aid in automated sorting processes by detecting non-rock objects in a conveyor belt system. |
|
|
| ## Model Details |
|
|
| - **Framework**: YOLOv8 |
| - **Training Data**: Custom rock vs. non-rock dataset |
| - **Purpose**: Object detection to differentiate between rocks and non-rock objects |
| - **License**: Apache 2.0 |
|
|
| ## Usage |
|
|
| To load the model: |
|
|
| ```python |
| from ultralytics import YOLO |
| |
| # Load the model |
| model = YOLO('hvc_model/best.pt') # Update this path based on your Hugging Face model path |
