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

Posted 2021-09-07 我把梦想卖了，换成了柴米油盐

说起来人工智能现在真是大热，之前也做过很多这方面的项目，譬如使用opencv识别车道线辅助驾驶，使用一些机器学习算法处理数字信号，使用BP神经网络处理信号或者其他的一些方面的应用，或者进行语音识别啊之类的，基本都是MATlab训练或者直接使用VC++ opencv 接口API或者其他模型借口这种，后来看网络上各种视频教程，这个tensorflow那个pytorch，这个YOLO那个SEq2seq，也是因为一些项目需要，把深度学习融入到了自己的项目中。
我有个朋友（是真的有个朋友
）就报了个深度学习的班，希望通过学习人工智能，拿下高薪，迎取白富美，走上人生巅峰，天天学啊学，这个模型，那个算法，在那个jupyter nootbook里照着老师的代码练阿练，最后学完了，感觉用处不大，和实际项目相差太远，谁会平时给你在这个记事本的环境里运行个模型有啥用，又不是让你搞科研，于是就问我，我说你可以自己重新写个模型，用java或者C++写，就可以在项目里用了，他说他不会，也太麻烦，好的吧，我说你可以在web框架里引用这个你训练过的模型，我让他找找网上的教程，他说网上都收费，我看他发过来的截图还真是，都不是什么富裕的人，哪有钱买这个，于是我给写个教程吧：

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

conda create -n flask python=3.7
新建一个环境

conda activate flask
激活这个环境

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

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

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

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/

就可以看到结果了：