用netron打開yolov8.onnx模型夷狰,可以發(fā)現(xiàn)模型的標簽信息已經(jīng)嵌入模型的元數(shù)據(jù)(Metadata)中了潭陪,使用代碼
import onnx
onnx_model = onnx.load("./yolov8n-cls.onnx")
# 獲取模型的元數(shù)據(jù)names
names = eval(onnx_model.metadata_props[9].value)
print(names[0])
tench
用netron打開yolov8.onnx模型夷狰,可以發(fā)現(xiàn)模型的標簽信息已經(jīng)嵌入模型的元數(shù)據(jù)(Metadata)中了潭陪,使用代碼
import onnx
onnx_model = onnx.load("./yolov8n-cls.onnx")
# 獲取模型的元數(shù)據(jù)names
names = eval(onnx_model.metadata_props[9].value)
print(names[0])
tench