别再手动P图了！用Python+Flask 5分钟搭建一个车牌图片生成API（支持蓝黄绿白黑牌）

5分钟构建车牌生成API：用Python+Flask打造高定制化图像服务

在自动化测试和图像处理领域，生成逼真的车牌图像是一个常见但容易被低估的需求。无论是用于车牌识别算法的训练数据增强，还是作为开发测试的模拟数据源，一个灵活的车牌生成API都能显著提升工作效率。传统的手动P图方式不仅耗时耗力，而且难以保证生成结果的多样性和合规性。

本文将带你从零开始，基于Python生态中的Flask框架和OpenCV图像处理库，构建一个支持多种车牌类型（蓝、黄、绿、白、黑）的RESTful API服务。与简单的接口调用不同，我们更关注工程化实现——包括参数校验、错误处理、代码结构优化以及生产环境部署方案，帮助开发者将这一功能无缝集成到自己的项目中。

1. 环境准备与项目初始化

1.1 基础依赖安装

开始前，确保你的Python环境版本≥3.6。我们推荐使用虚拟环境隔离项目依赖：

python -m venv plate_api_env source plate_api_env/bin/activate # Linux/macOS plate_api_env\Scripts\activate # Windows

安装核心依赖包：

pip install flask opencv-python numpy pillow

1.2 获取车牌生成基础库

我们将基于开源项目chinese_license_plate_generator进行二次开发。克隆项目仓库：

git clone https://gitee.com/leijd/chinese_license_plate_generator.git cd chinese_license_plate_generator

项目结构关键文件说明：

• plate_model/：车牌模板图像资源
• font_model/：车牌字体文件
• generate_multi_plate.py：核心生成逻辑

提示：建议fork原项目仓库以便自定义修改，如调整车牌样式或添加新类型

2. API服务核心架构设计

2.1 Flask应用骨架搭建

创建api_server.py作为服务入口文件：

from flask import Flask, request, jsonify import cv2 import base64 from generate_multi_plate import MultiPlateGenerator app = Flask(__name__) @app.route('/health') def health_check(): return jsonify({"status": "healthy"}) if __name__ == '__main__': app.run(host='0.0.0.0', port=5000, debug=True)

2.2 请求参数设计与校验

车牌生成API需要处理三类参数：

1. 车牌号码（plate_number）
2. 车牌颜色（plate_color）
3. 是否双层（double_layer）

参数校验表示例：

实现参数校验中间件：

def validate_plate_params(plate_number, plate_color): color_mapping = { 'blue': 'blue', 'yellow': 'yellow', 'green': 'green_car', 'white': 'white', 'black': 'black', 'b': 'blue', 'y': 'yellow', 'g': 'green', 'w': 'white', 'k': 'black' } if plate_color.lower() not in color_mapping: raise ValueError("Invalid plate color") # 车牌号正则校验（简化版） if not re.match(r'^[京津沪渝冀豫云辽黑湘皖鲁新苏浙赣鄂桂甘晋蒙陕吉闽贵粤青藏川宁琼][A-Z][A-HJ-NP-Z0-9]{4,5}$', plate_number): raise ValueError("Invalid plate number format") return color_mapping[plate_color.lower()]

3. 核心图像生成逻辑实现

3.1 集成原项目生成器

初始化车牌生成器并封装业务逻辑：

plate_generator = MultiPlateGenerator('plate_model', 'font_model') def generate_plate_image(plate_number, plate_color, double_layer=False): try: img = plate_generator.generate_plate_special( plate_number, plate_color, double_layer ) _, buffer = cv2.imencode('.jpg', img) return base64.b64encode(buffer).decode('utf-8') except Exception as e: app.logger.error(f"Generation failed: {str(e)}") raise

3.2 设计RESTful端点

实现主API端点，支持多种响应格式：

@app.route('/api/v1/plate', methods=['GET']) def generate_plate(): try: plate_number = request.args.get('number') plate_color = request.args.get('color', 'blue') double_layer = request.args.get('double', 'false').lower() == 'true' validated_color = validate_plate_params(plate_number, plate_color) image_base64 = generate_plate_image(plate_number, validated_color, double_layer) response_format = request.args.get('format', 'json') if response_format == 'image': return f'<img src="data:image/jpeg;base64,{image_base64}"/>' else: return jsonify({ "status": "success", "image": image_base64, "metadata": { "plate_number": plate_number, "plate_color": plate_color, "double_layer": double_layer } }) except ValueError as e: return jsonify({"status": "error", "message": str(e)}), 400 except Exception as e: return jsonify({"status": "error", "message": "Internal server error"}), 500

4. 生产环境部署优化

4.1 性能优化措施

当API面临高并发请求时，需要考虑以下优化：

• 对象复用：避免重复初始化生成器
• 缓存机制：对常见参数组合的结果缓存
• 异步处理：使用Celery处理生成任务

缓存实现示例：

from functools import lru_cache @lru_cache(maxsize=1000) def cached_generate(plate_number, plate_color, double_layer): return generate_plate_image(plate_number, plate_color, double_layer)

4.2 容器化部署

创建Dockerfile实现一键部署：

FROM python:3.9-slim WORKDIR /app COPY . . RUN pip install -r requirements.txt EXPOSE 5000 CMD ["gunicorn", "-w 4", "-b :5000", "api_server:app"]

构建并运行容器：

docker build -t plate-generator . docker run -d -p 5000:5000 plate-generator

4.3 API文档与测试

使用OpenAPI生成交互式文档：

from flask_swagger_ui import get_swaggerui_blueprint SWAGGER_URL = '/docs' API_URL = '/static/swagger.json' swaggerui_blueprint = get_swaggerui_blueprint( SWAGGER_URL, API_URL, config={'app_name': "Plate Generator API"} ) app.register_blueprint(swaggerui_blueprint, url_prefix=SWAGGER_URL)

典型测试用例：

# 蓝牌单层 curl "http://localhost:5000/api/v1/plate?number=京A12345&color=blue" # 黄牌双层 curl "http://localhost:5000/api/v1/plate?number=京A12345&color=yellow&double=true" # 获取JSON响应 curl "http://localhost:5000/api/v1/plate?number=京AD12345&color=green&format=json"

5. 高级功能扩展

5.1 批量生成模式

添加批量生成端点，支持一次请求多个车牌：

@app.route('/api/v1/plates/batch', methods=['POST']) def batch_generate(): try: requests_data = request.get_json() results = [] for item in requests_data: plate_number = item['number'] plate_color = item.get('color', 'blue') double_layer = item.get('double', False) validated_color = validate_plate_params(plate_number, plate_color) image_base64 = cached_generate(plate_number, validated_color, double_layer) results.append({ "plate_number": plate_number, "image": image_base64 }) return jsonify({"results": results}) except Exception as e: return jsonify({"error": str(e)}), 400

5.2 自定义样式支持

通过修改模板资源实现自定义样式：

1. 准备新的车牌模板图片放入plate_model/
2. 更新generate_multi_plate.py中的样式映射
3. 添加新的颜色枚举值

# 在generate_plate_image函数中添加自定义逻辑 if plate_color == 'custom': img = apply_custom_style(img, style='vip')

5.3 监控与日志

集成Prometheus监控指标：

from prometheus_flask_exporter import PrometheusMetrics metrics = PrometheusMetrics(app) metrics.info('plate_generator_info', 'Plate generator service info') # 添加自定义指标 generation_counter = metrics.counter( 'plate_generation_total', 'Total number of plate generations', labels={'color': lambda: request.args.get('color', 'unknown')} ) @app.route('/api/v1/plate') @generation_counter def generate_plate(): # 原有逻辑不变

日志配置示例：

import logging from logging.handlers import RotatingFileHandler handler = RotatingFileHandler('plate_api.log', maxBytes=10000, backupCount=3) handler.setLevel(logging.INFO) app.logger.addHandler(handler)