模型自帶的標(biāo)簽名
這樣的好處是：不用再為尋找標(biāo)簽文件費(fèi)時(shí)費(fèi)心了舞蔽。
從onnx模型讀取標(biāo)簽名的源代碼：
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)
運(yùn)行結(jié)果為：
{0: 'with_mask', 1: 'without_mask', 2: 'mask_weared_incorrect'}
很多同學(xué)問，那OpenVINO的IR模型支持這個(gè)屬性嗎？
答：2022.2版還沒有支持呜达，若支持，本文會(huì)更新
