xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 05:24:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update huggingface hub card template

Browse Source
This commit is contained in:
wa22 2024-06-07 16:22:42 +08:00
parent e161d70ff7
commit ea93d4f379
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/models/yolov10/card.py
View File

@ -1,6 +1,8 @@
card_template_text = """ card_template_text = """
--- ---
license: agpl-3.0 license: agpl-3.0
library_name: ultralytics
repo_url: https://github.com/THU-MIG/yolov10
tags: tags:
- object-detection - object-detection
- computer-vision - computer-vision

4
ultralytics/models/yolov10/model.py
View File

@ -7,9 +7,7 @@ from .train import YOLOv10DetectionTrainer
from huggingface_hub import PyTorchModelHubMixin from huggingface_hub import PyTorchModelHubMixin
from .card import card_template_text from .card import card_template_text
class YOLOv10(Model, PyTorchModelHubMixin, library_name="ultralytics", \ class YOLOv10(Model, PyTorchModelHubMixin, model_card_template=card_template_text):
repo_url="https://github.com/THU-MIG/yolov10", \
model_card_template=card_template_text):
def __init__(self, model="yolov10n.pt", task=None, verbose=False, def __init__(self, model="yolov10n.pt", task=None, verbose=False,
names=None): names=None):