从yolov8的模型元数据中获取标签信息
2024-01-05 本文已影响0人
LabVIEW_Python
用netron打开yolov8.onnx模型,可以发现模型的标签信息已经嵌入模型的元数据(Metadata)中了,使用代码
import onnx
onnx_model = onnx.load("./yolov8n-cls.onnx")
# 获取模型的元数据names
names = eval(onnx_model.metadata_props[9].value)
print(names[0])
tench
yolov8n-cls.onnx模型元数据