From b530a3004e8a409eeef323df9cbf8dde517ad22b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 31 Mar 2024 06:33:43 +0200 Subject: [PATCH] Define fallback `model.names` from Predictor (#9439) --- 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 b056c09e..ed03547c 100644 --- a/ultralytics/engine/model.py +++ b/ultralytics/engine/model.py @@ -733,7 +733,10 @@ class Model(nn.Module): """ 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: