lhollard commited on
Commit
969b244
1 Parent(s): ecd96b5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -3
README.md CHANGED
@@ -2,8 +2,23 @@
2
  tags:
3
  - model_hub_mixin
4
  - pytorch_model_hub_mixin
 
5
  ---
6
 
7
- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
- - Library: [More Information Needed]
9
- - Docs: [More Information Needed]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  tags:
3
  - model_hub_mixin
4
  - pytorch_model_hub_mixin
5
+ pipeline_tag: object-detection
6
  ---
7
 
8
+ # LeYOLO
9
+
10
+ ### LeYOLO Medium weights - HuggingFace integration
11
+
12
+ Github: https://github.com/LilianHollard/LeYOLO
13
+
14
+ Paper: https://arxiv.org/abs/2406.14239
15
+
16
+ Minimal example:
17
+
18
+ ```
19
+ #clone github
20
+ #inside LeYOLO folder, load LeYOLO weights (nano, small, medium or large)
21
+ model = YOLO.from_pretrained("lhollard/leyolo-medium")
22
+ #use ultralytics CLI and Python tools
23
+ results = model.val(data="ultralytics/cfg/datasets/coco8.yaml")
24
+ ```