1.ONNX介绍
ONNX(Open Neural Network Exchange)是一种开源格式,用于表示深度学习模型。它由Facebook和微软于2017年联合创建,目的是在不同的深度学习框架之间提供互操作性,使模型可以在多个平台和工具之间无缝转换和部署。ONNX定义了一种中间表示(IR),包括计算图、操作符和张量,能够描述从训练到推理的整个模型。
2.模型部署流程
扩展 图像可视化NETRON
直接上传导出的.onnx文件即可可视化模型
参考up主 同济子豪兄github地址如下:
【图像分类模型部署-Pytorch转ONNX】视频链接如下: