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.
1from ultralytics import YOLO
2
3# Load the model
4model = YOLO('hvc_model/best.pt') # Update this path based on your Hugging Face model path
