用OpenVINO转换ONNX模型

2021-06-14  本文已影响0人  LabVIEW_Python

1.什么是ONNX
Microsoft 和合作伙伴社区创建了 ONNX 作为表示机器学习模型的开放标准。 许多框架(包括 TensorFlow、PyTorch、飞桨等)中的模型都可以导出或转换为标准 ONNX 格式。 模型采用 ONNX 格式后,可在各种平台和设备上运行。

ONNX支持的框架

2. 从ONNX模型库中下载AlexNet模型
下载链接:https://github.com/onnx/models/blob/master/vision/classification/alexnet/model/bvlcalexnet-9.onnx

3. 用Netron查看AlexNet模型结构

bvlcalexnet-9.png

4. 用OpenVINO将ONNX模型转化为IR模型

python mo_onnx.py --input_model d:\training\bvlcalexnet-9.onnx --output_dir d:\training

转换ONNX模型
上一篇 下一篇

猜你喜欢

热点阅读