图像分类模型部署-Pytorch转ONNX

[视频作者] 同济子豪兄

[视频时长] 16:41

[视频类型] 计算机技术

两天搞定图像分类毕业设计 代码教程：https://github.com/TommyZihao/Train_Custom_Dataset 前面的视频中，我们已构建了自己的30类图像分类数据集，用Pytorch迁移学习训练得到自己的图像分类模型。但原生Pytorch模型的推理预测速度慢，且无法完美兼容适配在各种终端硬件设备上。 本讲视频，分别演示将Pytorch模型库预训练ImageNet模型，和自己训练得到的水果分类模型，转换为机器学习通用中间格式ONNX，为后续ONNX Runtime、TensorR