keffy/hvc_model

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