员工卡照片一键生成器～

from flask import Flask, request, send_file,render_template import cv2 import numpy as np from io import BytesIO from flask_cors import CORS app = Flask(__name__) CORS(app) @app.route('/', methods=['GET']) def home(): return render_template("index.html") @app.route('/process_image', methods=['POST']) def process_image(): try: # 获取上传的图片 file = request.files['image'] img = cv2.imdecode(np.frombuffer(file.read(), np.uint8), cv2.IMREAD_COLOR) # 获取参数 width = int(request.form.get('width', img.shape[1])) height = int(request.form.get('height', img.shape[0])) file_size = int(request.form.get('fileSize', 100)) background_color = request.form.get('backgroundColor', 'white') # 背景去除 mask = np.zeros(img.shape[:2], np.uint8) bgdModel = np.zeros((1, 65), np.float64) fgdModel = np.zeros((1, 65), np.float64) rect = (10, 10, img.shape[1] - 20, img.shape[0] - 20) cv2.grabCut(img, mask, rect, bgdModel, fgdModel, 5, cv2.GC_INIT_WITH_RECT) bin_mask = np.where((mask == 2) | (mask == 0), 0, 255).astype('uint8') alpha = cv2.cvtColor(bin_mask, cv2.COLOR_GRAY2BGR) rgba = cv2.merge([img, alpha]) # 调整大小 resized_img = cv2.resize(rgba, (width, height)) # 替换背景颜色 color_map = { 'white': [255, 255, 255], 'blue': [0, 0, 255], 'red': [255, 0, 0] } bg_color = color_map.get(background_color, [255, 255, 255]) bg = np.full((height, width, 3), bg_color, dtype=np.uint8) bg = cv2.cvtColor(bg, cv2.COLOR_BGR2BGRA) bg[:, :, 3] = 255 mask = resized_img[:, :, 3] == 0 bg[mask] = resized_img[mask] # 调整文件大小 quality = 100 while True: is_success, buffer = cv2.imencode('.jpg', bg, [cv2.IMWRITE_JPEG_QUALITY, quality]) if is_success: file_size_kb = len(buffer) / 1024 if file_size_kb <= file_size or quality < 10: break quality -= 10 else: break # 返回处理后的图片 img_bytes = BytesIO(buffer) img_bytes.seek(0) return send_file(img_bytes, mimetype='image/jpeg') except Exception as e: return str(e), 500 if __name__ == '__main__': app.run(debug=True)