README.md

metadata

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:

from ultralytics import YOLO

# Load the model
model = YOLO('hvc_model/best.pt')  # Update this path based on your Hugging Face model path