From 58a05f8e708f1ebcb8ce8d36ae7eee7a49aaf428 Mon Sep 17 00:00:00 2001 From: Jamjamjon <51357717+jamjamjon@users.noreply.github.com> Date: Fri, 15 Mar 2024 02:46:56 +0800 Subject: [PATCH] Fix `names` bug when exporting YOLOv8-World to ONNX (#8941) Co-authored-by: Glenn Jocher --- ultralytics/models/yolo/model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ultralytics/models/yolo/model.py b/ultralytics/models/yolo/model.py index 3a6445f4..1bd38d3b 100644 --- a/ultralytics/models/yolo/model.py +++ b/ultralytics/models/yolo/model.py @@ -71,8 +71,9 @@ class YOLOWorld(Model): """ super().__init__(model=model, task="detect") - # Assign default COCO class names - self.model.names = yaml_load(ROOT / "cfg/datasets/coco8.yaml").get("names") + # Assign default COCO class names when there are no custom names + if not hasattr(self.model, "names"): + self.model.names = yaml_load(ROOT / "cfg/datasets/coco8.yaml").get("names") @property def task_map(self):