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

Changed model.names return type hint from list to dict[int, str] (#17402)

Browse source
This commit is contained in:
Kilian Schnelle 2024-11-08 16:19:46 +01:00 • committed by GitHub
parent 84bfe3c2e4
commit 95368cbe54
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ultralytics/engine/model.py
View file

@ -2,7 +2,7 @@
import inspect import inspect
from pathlib import Path from pathlib import Path
from typing import List, Union from typing import Dict, List, Union
import numpy as np import numpy as np
import torch import torch
@ -881,7 +881,7 @@ class Model(nn.Module):
return self return self
@property @property
def names(self) -> list: def names(self) -> Dict[int, str]:
""" """
Retrieves the class names associated with the loaded model. Retrieves the class names associated with the loaded model.