从onnx模型的Meta数据中获取标签名

用Netron打开onnx模型，可以看到标签名，如下图所示：

模型自带的标签名

这样的好处是：不用再为寻找标签文件费时费心了。

从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版还没有支持，若支持，本文会更新