美文网首页
用OpenVINO转换ONNX模型

用OpenVINO转换ONNX模型

作者: LabVIEW_Python | 来源:发表于2021-06-14 18:59 被阅读0次

    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模型

    相关文章

      网友评论

          本文标题:用OpenVINO转换ONNX模型

          本文链接:https://www.haomeiwen.com/subject/tfuoeltx.html