pytorch图像分类实战 之模型部署
前言
第六次任务是模型部署。代码教程：https://github.com/TommyZihao/Train_Custom_Dataset 。
实验效果
总结
这次学习主要是对于模型部署进行了初步了解，并根据教程给的ONNX Runtime部署方法做了一边实验，看了看代码实现，对于模型部署有了初步认识。ONNX Runtime是一个跨平台推理和训练机器学习加速器，是微软开发的一个深度学习推理引擎，就是tensorflow/pytorch/paddle等深度学习框架训练的模型都可以转换为ONNX格式，然后ONNX格式模型可以通过ONNX runtime组件实现模型的推理预测并加速，从而实现不基于原来框架的模型部署。
本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。
THE END
二维码