|
--- |
|
tags: |
|
- model_hub_mixin |
|
- pytorch_model_hub_mixin |
|
pipeline_tag: object-detection |
|
--- |
|
|
|
# LeYOLO |
|
|
|
### LeYOLO Medium weights - HuggingFace integration |
|
|
|
Github: https://github.com/LilianHollard/LeYOLO |
|
|
|
Paper: https://arxiv.org/abs/2406.14239 |
|
|
|
Minimal example: |
|
|
|
``` |
|
#clone github |
|
#inside LeYOLO folder, load LeYOLO weights (nano, small, medium or large) |
|
model = YOLO.from_pretrained("lhollard/leyolo-medium") |
|
#use ultralytics CLI and Python tools |
|
results = model.val(data="ultralytics/cfg/datasets/coco8.yaml") |
|
``` |
