HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Define export formats model.names attribute (#9474)

Browse source
This commit is contained in:
Glenn Jocher 2024-04-01 18:53:44 +02:00 • committed by GitHub
parent 80c59e2461
commit 2cee8893d9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ultralytics/engine/model.py
View file

@ -735,7 +735,10 @@ class Model(nn.Module):
if hasattr(self.model, "names"): if hasattr(self.model, "names"):
return check_class_names(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 return self.predictor.model.names
@property @property