Flask web服务部署深度学习模型YOLO

说起来人工智能现在真是大热,之前也做过很多这方面的项目,譬如使用opencv识别车道线辅助驾驶,使用一些机器学习算法处理数字信号,使用BP神经网络处理信号或者其他的一些方面的应用,或者进行语音识别啊之类的,基本都是MATlab训练或者直接使用VC++ opencv 接口API或者其他模型借口这种,后来看网络上各种视频教程,这个tensorflow那个pytorch,这个YOLO那个SEq2seq,也是因为一些项目需要,把深度学习融入到了自己的项目中。

Flask web服务部署深度学习模型YOLO
)就报了个深度学习的班,希望通过学习人工智能,拿下高薪,迎取白富美,走上人生巅峰,天天学啊学,这个模型,那个算法,在那个jupyter nootbook里照着老师的代码练阿练,最后学完了,感觉用处不大,和实际项目相差太远,谁会平时给你在这个记事本的环境里运行个模型有啥用,又不是让你搞科研,于是就问我,我说你可以自己重新写个模型,用java或者C++写,就可以在项目里用了,他说他不会,也太麻烦,好的吧,我说你可以在web框架里引用这个你训练过的模型,我让他找找网上的教程,他说网上都收费,我看他发过来的截图还真是,都不是什么富裕的人,哪有钱买这个,于是我给写个教程吧:

Flask web服务部署深度学习模型YOLO

Flask web服务部署深度学习模型YOLO

我就教他点简单的吧,好在我这个web编程方面还算过关,就给他用FLASk这个web框架给他演示一遍,在这里我要感谢这个github的分享:https://github.com/xugaoxiang/yolov5-flask,写的很不错。
首先你要下载一个pycharm,anaconda安装好,在anaconda prompt里输入

Flask web服务部署深度学习模型YOLO
conda create -n flask python=3.7
新建一个环境

conda activate flask
激活这个环境

pip install flask opencv-python
安装我们的web框架flask

然后打开pycharm后,file-setting里选择自己的环境;

Flask web服务部署深度学习模型YOLO
Flask web服务部署深度学习模型YOLO

在这个工作台里安装我们需要的包,

pip install opencv-python~=4.4.0.44
pip install matplotlib~=3.3.3
pip install numpy~=1.19.2
pip install torch~=1.6.0+cu101
pip install torchvision~=0.7.0+cu101
pip install scipy~=1.5.4
pip install tqdm~=4.54.1
pip install pyyaml~=5.3.1
pip install pillow~=8.0.0
pip install flask~=1.1.2

使用pip安装python模块时报错如下
Exception:
Traceback (most recent call last):
File “C:\ProgramData\Anaconda3\lib\shutil.py”, line 544, in move
os.rename(src, real_dst)
PermissionError: [WinError 5] 拒绝访问。

解决方法

可能会出现:ERROR: Could not find a version that satisfies the requirement torch,解决方案在我另一篇博客:ERROR: Could not find a version that satisfies the requirement torch

然后运行在pycharm工作台运行
python app.py

访问;http://192.168.2.131:5001/

就可以看到结果了:

Flask web服务部署深度学习模型YOLO

Original: https://blog.csdn.net/leva345/article/details/119835177
Author: Mr Robot
Title: Flask web服务部署深度学习模型YOLO

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/746937/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球