From 2cee8893d98210560a883dae08b8ad41eb04b6b5 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 1 Apr 2024 18:53:44 +0200
Subject: [PATCH] Define export formats `model.names` attribute (#9474)

---
 ultralytics/engine/model.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/ultralytics/engine/model.py b/ultralytics/engine/model.py
index cae5023b..2c178016 100644
--- a/ultralytics/engine/model.py
+++ b/ultralytics/engine/model.py
@@ -735,7 +735,10 @@ class Model(nn.Module):
 
         if hasattr(self.model, "names"):
             return check_class_names(self.model.names)
-        elif self.predictor:
+        else:
+            if not self.predictor:  # export formats will not have predictor defined until predict() is called
+                self.predictor = self._smart_load("predictor")(overrides=self.overrides, _callbacks=self.callbacks)
+                self.predictor.setup_model(model=self.model, verbose=False)
             return self.predictor.model.names
 
     @property