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

Fix names bug when exporting YOLOv8-World to ONNX (#8941)

Browse Source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Jamjamjon 2024-03-15 02:46:56 +08:00 committed by GitHub
parent 5c1277113b
commit 58a05f8e70
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ultralytics/models/yolo/model.py
View File

@ -71,8 +71,9 @@ class YOLOWorld(Model):
""" """
super().__init__(model=model, task="detect") super().__init__(model=model, task="detect")
# Assign default COCO class names # Assign default COCO class names when there are no custom names
self.model.names = yaml_load(ROOT / "cfg/datasets/coco8.yaml").get("names") if not hasattr(self.model, "names"):
self.model.names = yaml_load(ROOT / "cfg/datasets/coco8.yaml").get("names")
@property @property
def task_map(self): def task_map(self):