### 如何使用 Flask 部署 YOLOv5 模型 为了在 WEB 应用程序中部署 YOLOv5 模型，可以利用 Python 的轻量级框架 Flask 构建后端服务。以下是详细的说明以及示例代码。 #### 后端环境准备 首先需要安装必要的依赖库，包括 `Flask` 和 `torchvision` 等支持 YOLOv5 运行的工具包。可以通过以下命令完成安装： ```bash pip install flask torch torchvision pyyaml opencv-python-headless ``` #### 下载并加载 YOLOv5 模型 YOLOv5 提供了预训练权重文件，可以直接从其 GitHub 仓库下载所需模型及其权重文件[^3]。例如，可以选择官方推荐的小规模模型 yolo5s.pt 来测试性能。 加载模型的方法如下所示： ```python import torch def load_model(): model = torch.hub.load('ultralytics/yolov5', 'custom', path='path/to/best.pt') # 加载自定义模型 return model ``` #### 创建 Flask API 接口 构建一个简单的 RESTful API，允许客户端上传图片并通过 POST 请求调用目标检测功能。下面是一个基本的应用实例： ```python from flask import Flask, request, jsonify import cv2 import numpy as np app = Flask(__name__) model = None @app.before_first_request def initialize(): global model model = load_model() @app.route('/detect', methods=['POST']) def detect_objects(): if 'file' not in request.files: return jsonify({'error': 'No file part'}), 400 file = request.files['file'] if file.filename == '': return jsonify({'error': 'No selected file'}), 400 try: image_bytes = file.read() img_nparr = np.frombuffer(image_bytes, dtype=np.uint8) frame = cv2.imdecode(img_nparr, flags=cv2.IMREAD_COLOR) results = model(frame) # 使用模型进行推理 detections = results.pandas().xyxy[0].to_dict(orient="records") response_data = { "detections": detections, "image_shape": {"height": frame.shape[0], "width": frame.shape[1]} } return jsonify(response_data), 200 except Exception as e: return jsonify({"error": str(e)}), 500 if __name__ == '__main__': app.run(host='0.0.0.0', port=5000) ``` 此部分实现了接收图像输入的功能，并返回检测到的对象列表及位置坐标信息[^1]。 #### 前端交互设计 (可选) 如果希望进一步增强用户体验，则可以在前端页面加入 HTML 表单来提交待处理的照片；或者采用 JavaScript 实现异步 AJAX 调用来简化操作流程。这里仅给出最基础的形式作为参考： ```html <!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Object Detection</title> </head> <body> <h1>Upload an Image for Object Detection</h1> <form action="/detect" method="post" enctype="multipart/form-data"> Select image to upload: <input type="file" name="file"><br><br> <button type="submit">Detect Objects</button> </form> </body> </html> ``` 以上即为完整的基于 Flask 的 YOLOv5 目标检测系统的搭建过程概述[^2]。 ---