基于flask和网页端部署yolo自训练模型
关于yolov5模型的flask的部署, 需要通过torch.hub.load 加载yolov5 自定义的模型,source需要设置local。关于models里面的模型,一个是官方yolov5s.pt,另外一个是自训练的口罩模型,检测是否戴有口罩。关于 http://127.0.0.1:5000/v1 接口,返回的是检测结果。检测的图片会保存在static中的forest文件夹。在torch的环境,运行app.py。也可以通过curl对图片进行上传访问。本项目支持运行自己的自训练模型的模型。......