YOLOv5-Flask 项目使用教程

YOLOv5-Flask 项目使用教程

yolov5-flaskOfficial implementation at https://github.com/ultralytics/yolov5/tree/master/utils/flask_rest_api项目地址:https://gitcode.com/gh_mirrors/yo/yolov5-flask

1. 项目的目录结构及介绍

robmarkcole/yolov5-flask/
├── docs/
├── static/
├── templates/
├── tests/
├── .gitignore
├── Dockerfile
├── LICENSE
├── README.md
├── requirements.txt
├── restapi.py
└── webapp.py

• docs/: 存放项目文档文件。
• static/: 存放静态文件，如CSS、JavaScript文件等。
• templates/: 存放HTML模板文件。
• tests/: 存放测试文件。
• .gitignore: Git忽略文件配置。
• Dockerfile: Docker容器配置文件。
• LICENSE: 项目许可证文件。
• README.md: 项目说明文档。
• requirements.txt: 项目依赖包列表。
• restapi.py: REST API接口文件。
• webapp.py: Web应用启动文件。

2. 项目的启动文件介绍

webapp.py

webapp.py 是项目的启动文件，负责启动Flask应用并加载YOLOv5模型。以下是文件的主要内容：

from flask import Flask, render_template, Response
import cv2
import torch

app = Flask(__name__)

model = torch.hub.load('ultralytics/yolov5', 'yolov5s')

@app.route('/')
def index():
    return render_template('index.html')

def gen_frames():
    camera = cv2.VideoCapture(0)
    while True:
        success, frame = camera.read()
        if not success:
            break
        else:
            results = model(frame)
            ret, buffer = cv2.imencode('.jpg', frame)
            frame = buffer.tobytes()
            yield (b'--frame\r\n'
                   b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')

@app.route('/video_feed')
def video_feed():
    return Response(gen_frames(), mimetype='multipart/x-mixed-replace; boundary=frame')

if __name__ == '__main__':
    app.run(debug=True)

restapi.py

restapi.py 文件定义了REST API接口，用于通过HTTP请求调用YOLOv5模型进行目标检测。

from flask import Flask, request, jsonify
import torch

app = Flask(__name__)

model = torch.hub.load('ultralytics/yolov5', 'yolov5s')

@app.route('/detect', methods=['POST'])
def detect():
    image = request.files['image']
    results = model(image)
    return jsonify(results.pandas().xyxy[0].to_dict())

if __name__ == '__main__':
    app.run(debug=True)

3. 项目的配置文件介绍

requirements.txt

requirements.txt 文件列出了项目运行所需的Python依赖包及其版本。

flask
torch
opencv-python

Dockerfile

Dockerfile 文件用于构建Docker容器，确保项目在不同环境中的一致性。

FROM python:3.8-slim

WORKDIR /app

COPY requirements.txt requirements.txt
RUN pip install -r requirements.txt

COPY . .

CMD ["python", "webapp.py"]

.gitignore

.gitignore 文件用于指定Git版本控制系统中需要忽略的文件和目录。

__pycache__/
*.pyc
*.pyo
*.db
.env

通过以上介绍，您可以更好地理解和使用 robmarkcole/yolov5-flask 项目。希望这份教程对您有所帮助！

yolov5-flaskOfficial implementation at https://github.com/ultralytics/yolov5/tree/master/utils/flask_rest_api项目地址:https://gitcode.com/gh_mirrors/yo/yolov5-flask