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

pull/17384/head^2
Kilian Schnelle 6 days ago committed by GitHub
parent 84bfe3c2e4
commit 95368cbe54
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 4
      ultralytics/engine/model.py

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

Loading…
Cancel
Save