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
Downloads last month
3
Inference Examples
Unable to determine this model's library. Check the docs .

Evaluation results