|
--- |
|
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 |