Egg-Detection / README.md
afshin-dini's picture
Update README.md
20f39de verified
|
raw
history blame
790 Bytes
---
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](https://huggingface.co/datasets/afshin-dini/Egg-Detection).
## Code
The complete code can be found [here](https://github.com/afshindini/Deep-Egg-Detection-and-Counter).
## How to use
One can use the model within his/her code with the following commands:
```shell
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")
```