用netron打開yolov8.onnx模型,可以發(fā)現(xiàn)模型的標(biāo)簽信息已經(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

yolov8n-cls.onnx模型元數(shù)據(jù)