YOLOv5車牌識別實戰教學(八)Web應用與API開發

2023-04-04 06:01:01

Flask框架介紹

Flask是一個輕量級的Python Web框架，它非常適合開發小型到中型的Web應用。Flask提供了基本的路由、模板渲染和表單處理功能。

在這個教學中，我們將使用Flask來實現一個簡單的車牌識別Web應用。

首先，確保你已經安裝了Flask。如果沒有，請使用以下命令安裝：

pip install flask

實現車牌識別Web應用

建立基本的Flask應用

首先，我們需要建立一個基本的Flask應用。建立一個名為app.py的檔案，並新增以下程式碼：

from flask import Flask, render_template, request, redirect, url_for
 
app = Flask(__name__)
 
@app.route('/')
def index():
    return render_template('index.html')
 
if __name__ == '__main__':
    app.run(debug=True)

在這個程式碼中，我們首先匯入了必要的Flask模組。然後，我們定義了一個名為index的檢視函數，並使用@app.route('/')裝飾器將其與根URL（/）關聯。最後，我們在__main__中執行Flask應用。

新增HTML模板

接下來，我們需要建立一個HTML模板檔案。在專案資料夾中建立一個名為templates的資料夾，並在其中建立一個名為index.html的檔案。

在index.html中新增以下程式碼：

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>YOLOv5 License Plate Recognition</title>
</head>
<body>
    <h1>YOLOv5 License Plate Recognition</h1>
    <form action="{{ url_for('upload') }}" method="POST" enctype="multipart/form-data">
        <input type="file" name="image" required>
        <button type="submit">Submit</button>
    </form>
</body>
</html>

在這個模板中，我們建立了一個簡單的表單，用於上傳車輛圖片。表單的action屬性設定為{{ url_for('upload') }}，這將在後續步驟中與上傳圖片的檢視函數關聯。

實現圖片上傳與車牌識別

現在，我們需要實現上傳圖片和進行車牌識別的檢視函數。在app.py中新增以下程式碼：

import os
import cv2
from werkzeug.utils import secure_filename
from license_plate_recognition import process_image
 
UPLOAD_FOLDER = 'uploads'
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 
def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

接下來，在app.py中實現上傳圖片和進行車牌識別的檢視函數：

@app.route('/upload', methods=['POST'])
def upload():
    if 'image' not in request.files:
        return redirect(request.url)
    
    file = request.files['image']
    if file.filename == '':
        return redirect(request.url)
    
    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
        filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
        file.save(filepath)
 
        # 車牌識別
        image = cv2.imread(filepath)
        results = process_image(image)
 
        # 渲染結果
        return render_template('results.html', results=results)
 
    return redirect(request.url)

在這個檢視函數中，我們首先檢查請求中是否包含圖片檔案。如果包含，則檢查檔案型別是否允許。接下來，我們將圖片儲存到UPLOAD_FOLDER中，並使用我們之前實現的process_image()函數對其進行車牌識別。最後，我們將識別結果渲染到一個新的HTML模板results.html。

建立車牌識別API

除了建立一個Web應用外，我們還可以為其他開發者提供一個車牌識別API。這樣，其他人可以輕鬆地將我們的車牌識別功能整合到他們的應用中。

在app.py中新增以下程式碼：

from flask import jsonify
 
@app.route('/api/recognize', methods=['POST'])
def recognize_api():
    if 'image' not in request.files:
        return jsonify({'error': 'No image provided'}), 400
 
    file = request.files['image']
    if file.filename == '':
        return jsonify({'error': 'No image provided'}), 400
 
    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
        filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
        file.save(filepath)
 
        # 車牌識別
        image = cv2.imread(filepath)
        results = process_image(image)
 
        # 返回識別結果
        return jsonify({'results': results})
 
    return jsonify({'error': 'Invalid file type'}), 400

在這個檢視函數中，我們與Web應用的處理邏輯類似，但是將結果以JSON格式返回，而不是渲染HTML模板。

擴充套件功能

在基本的Web應用和API基礎上，你還可以根據需求新增更多功能，例如：

資料庫整合

將識別結果儲存到資料庫中，以便進行進一步的分析和管理。可以選擇MySQL、PostgreSQL、MongoDB等資料庫進行整合。

使用者認證與授權

為API新增使用者認證和授權功能，確保只有授權的使用者才能使用車牌識別API。

實時視訊流處理

擴充套件Web應用和API，使其支援實時視訊流處理。這樣，使用者可以實時檢視和分析車牌識別結果。

車輛資訊查詢

根據識別到的車牌號碼，查詢車輛的詳細資訊，如車型、車主、註冊日期等。

總結

在本教學中，我們學習瞭如何使用Flask框架為YOLOv5車牌識別系統開發Web應用和API。我們實現了一個簡單的車牌識別Web應用，併為其他開發者提供了一個易於整合的車牌識別API。通過將車牌識別技術部署到各種應用場景，我們可以為智慧交通、停車場管理等提供便利。希望本教學能為你在實際專案中應用車牌識別技術提供幫助。如有任何問題或建議，請在評論區交流。

到此這篇關於YOLOv5車牌識別實戰教學(八)Web應用與API開發的文章就介紹到這了,更多相關YOLOv5車牌識別Web應用與API開發內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！