【6】图像分类部署
文章目录
- 前言
- 一、将pytorch模型转为ONNX
- 二、本地终端部署
- 2.1. ONNX Runtime部署
- 2.2. pytorch模型部署(补充)
- 三、使用flask的web网页部署
- 四、微信小程序部署
- 五、使用pyqt界面化部署
- 总结
前言
包括将训练好的模型部署在本地终端、web端、小程序上、qt界面化。
三叶青图像识别研究简概
一、将pytorch模型转为ONNX
因为mobilenet模型在GPU上训练得到,要使其在CPU 上进行模型推理,需要确保模型的权重被正确地加载到CPU上。
然后再将获得的在CPU上进行推理的pytorch模型转化为ONNX模型。并可使用Netron可视化模型结构。
二、本地终端部署
2.1. ONNX Runtime部署
使用推理引擎ONNX Runtime,读取onnx格式的模型文件,对单张图像、视频进行预测。
2.2. pytorch模型部署(补充)
不将.pth模型文件转化为onnx模型文件,而是直接部署pytorch模型文件
三、使用flask的web网页部署
flask+bootstrap+jquery+mysql搭建三叶青在线识别网站,使用nginx+gunicorn将网站部署在腾讯云上,配置SSL证书。网站地址:https://www.whtuu.cn
参考我的另外一篇文章:使用flask的web网页部署介绍
四、微信小程序部署
利用uniapp+微信开发者工具+flask后端开发了三叶青图像识别微信小程序。
参考我的另外一篇文章:微信小程序部署
五、使用pyqt界面化部署
pyqt+opencv开发的图像识别qt界面
目前共有五个主要界面:软件介绍界面、省份识别、浙产识别、产地识别界面、以及自定义识别页面。
参考我的另外一篇文章:使用pyqt界面化部署
总结
本文主要介绍了如何将pytorch模型转为ONNX模型文件以及如何将图像识别模型部署,包括部署在本地、部署在web网页、小程序、qt界面部署。
2024/6/13