Egg-Detection / README.md
afshin-dini's picture
Update README.md
20f39de verified
metadata
license: mit
datasets:
  - afshin-dini/Egg-Detection
base_model:
  - Ultralytics/YOLOv8
pipeline_tag: object-detection
tags:
  - object-detection
  - YOLO
  - Egg-Detection
library_name: pytorch

Description

This model is a YOLO-based model which is trained on Egg Detection Dataset.

Code

The complete code can be found here.

How to use

One can use the model within his/her code with the following commands:

from huggingface_hub import hf_hub_download
from ultralytics import YOLO

model_path = hf_hub_download(repo_id="afshin-dini/Egg-Detection", filename="model/egg_detector.pt")
model = YOLO(model_path)
result = model("path/to/image")