从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模型元数据
上一篇 下一篇

猜你喜欢

热点阅读