fastapi封装yolov5自训练模型

303 阅读1分钟

简介:本次用到的模型为基于yolov5自训练的口罩检测识别模型,测试工具是postman,如有问题请留言

1、必备资源

github下载yolov5源码,下载yolov5s.pt模型文件

使用yolov5s.pt模型或者自己基于yolov5源码训练的模型

下载postman工具(fastapi进行模型测试时使用)

2、安装必备依赖
其中unvicron是基于 **uvloop** 和 httptools 构建的非常快速的 ASGI 服务器

pip install fastapi
pip install unvicron
3、找到yolov5源码中的detect.py文件,然后修改其中的如图所示:

在run方法最后新增代码,返回推理完成后图片的路径

image.png

4、新建webinfer.py文件,来搭建fastapi框架进行模型的推理,代码如下:

5、执行webinfer.py文件,得到如图所示即为正常:

image.png

6、打开postman工具,配置参数,根据代码中的信息,填写对应参数(source对应的是测试图片的路径)

image.png 点击发送得到最后结果

image.png 后端将会显示200 ok表示响应成功

image.png