Skip to content

Commit aa26f2f

Browse files
committed
extract cls names list
1 parent d063a5c commit aa26f2f

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

roboflow/core/version.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -307,11 +307,17 @@ def deploy(self, model_type: str, model_path: str) -> None:
307307
"The ultralytics python package is required to deploy yolov8 models. Please install it with `pip install ultralytics`"
308308
)
309309

310+
class_names = []
311+
for i, val in enumerate(m["model"].names):
312+
class_names.append((val, m["model"].names[val]))
313+
class_names.sort(key=lambda x: x[0])
314+
class_names = [x[1] for x in class_names]
315+
310316
# add logic to save torch state dict safely
311317
if model_type == "yolov8":
312318
model = torch.load(model_path + "weights/best.pt")
313319
model_artifacts = {
314-
"names": model["model"].names,
320+
"names": class_names,
315321
"yaml": model["model"].yaml,
316322
"nc": model["model"].nc,
317323
"args": {

0 commit comments

Comments
 (0)