Update README.md

README.md

@@ -1,9 +1,49 @@
 ---
+license: agpl-3.0
 tags:
-- pytorch_model_hub_mixin
-- model_hub_mixin
+- object-detection
+- computer-vision
+- yolov10
+datasets:
+- detection-datasets/coco
 ---
 
-This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
-- Library: [More Information Needed]
-- Docs: [More Information Needed]
+### Model Description
+[YOLOv10: Real-Time End-to-End Object Detection](https://arxiv.org/abs/2405.14458v1)
+
+- arXiv: https://arxiv.org/abs/2405.14458v1
+- github: https://github.com/THU-MIG/yolov10
+
+### Installation
+```
+pip install supervision git+https://github.com/THU-MIG/yolov10.git
+```
+
+### Yolov10 Training/Validation/Prediction
+```python
+from ultralytics import YOLOv10
+import cv2
+IMAGE_PATH = 'dog.jpeg'
+
+model = YOLOv10.from_pretrained('jameslahm/yolov10s')
+# Training
+model.train(...)
+# after training, one can push to the hub
+model.push_to_hub("your-hf-username/yolov10-finetuned")
+
+# Validation
+model.val(...)
+
+# Prediction
+model.predict(IMAGE_PATH, show=True)
+```
+
+### BibTeX Entry and Citation Info
+ ```
+@article{wang2024yolov10,
+  title={YOLOv10: Real-Time End-to-End Object Detection},
+  author={Wang, Ao and Chen, Hui and Liu, Lihao and Chen, Kai and Lin, Zijia and Han, Jungong and Ding, Guiguang},
+  journal={arXiv preprint arXiv:2405.14458},
+  year={2024}
+}
+```