这样的好处是:不用再为寻找标签文件费时费心了。
从onnx模型读取标签名的源代码:
import onnxruntime
# Load Label from Model's meta data
session = onnxruntime.InferenceSession("mask_yolov5.onnx", providers=['CPUExecutionProvider'])
meta = session.get_modelmeta().custom_metadata_map # metadata
label_names = eval(meta['names'])
print(label_names)
运行结果为:
{0: 'with_mask', 1: 'without_mask', 2: 'mask_weared_incorrect'}
很多同学问,那OpenVINO的IR模型支持这个属性吗?
答:2022.2版还没有支持,若支持,本文会更新