Define fallback model.names from Predictor (ultralytics#9439)

hmurari · Apr 17, 2024 · 37fbb06 · 37fbb06
1 parent 8d61dc9
commit 37fbb06
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ultralytics/engine/model.py b/ultralytics/engine/model.py
@@ -733,7 +733,10 @@ def names(self) -> list:
         """
         from ultralytics.nn.autobackend import check_class_names
 
-        return check_class_names(self.model.names) if hasattr(self.model, "names") else None
+        if hasattr(self.model, "names"):
+            return check_class_names(self.model.names)
+        elif self.predictor:
+            return self.predictor.model.names
 
     @property
     def device(self) -> torch.device: