|
|
--- |
|
|
license: mit |
|
|
datasets: |
|
|
- afshin-dini/Egg-Detection |
|
|
base_model: |
|
|
- Ultralytics/YOLOv8 |
|
|
pipeline_tag: object-detection |
|
|
tags: |
|
|
- object-detection |
|
|
- yolo |
|
|
- pytorch |
|
|
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") |
|
|
``` |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
