LeYOLO, New Scalable and Efficient CNN Architecture for Object Detection
Paper • 2406.14239 • Published
How to use lhollard/leyolo-small with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("object-detection", model="lhollard/leyolo-small") # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("lhollard/leyolo-small", dtype="auto")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-small")
#use ultralytics CLI and Python tools
results = model.val(data="ultralytics/cfg/datasets/coco8.yaml")