ADD file via upload

advanced/license/license_recog.py

@@ -0,0 +1,144 @@
+import tkinter as tk
+from tkinter import ttk, filedialog, messagebox
+from tkinter import Toplevel
+from PIL import Image, ImageTk
+from ttkbootstrap import Style
+import numpy as np
+import cv2
+import os
+from .image_operations import locate_license_plate, recognize_characters
+
+edge = None
+
+processed_win = None
+
+def select_image(root, img_container):
+    global edge
+
+    img_path = filedialog.askopenfilename(filetypes=[("Image files", "*.jpg;*.png;*.jpeg;*.bmp")])
+    if img_path:
+        img_path_fixed = os.path.normpath(img_path)
+        img_container['file_path'] = img_path_fixed
+
+        src_temp = cv2.imdecode(np.fromfile(img_path_fixed, dtype=np.uint8), cv2.IMREAD_UNCHANGED)
+        if src_temp is None:
+            messagebox.showerror("错误", "无法读取图片，请选择有效的图片路径")
+            return
+
+        src = cv2.cvtColor(src_temp, cv2.COLOR_BGR2RGB)
+        img_container['image'] = src
+
+        if 'img_label' not in img_container or not isinstance(img_container['img_label'], tk.Label):
+            img_container['img_label'] = tk.Label(root)
+            img_container['img_label'].pack(side=tk.TOP, pady=10)
+
+        img = Image.fromarray(src)
+        img.thumbnail((200, 200))
+        img_tk = ImageTk.PhotoImage(img)
+        img_container['img_label'].configure(image=img_tk)
+        img_container['img_label'].image = img_tk
+
+        edge = Image.fromarray(src)
+    else:
+        messagebox.showerror("错误", "没有选择图片路径")
+
+def show_selected_image(root, img_container):
+    if 'file_path' in img_container:
+        img_path = img_container['file_path']
+        img = Image.open(img_path)
+        img.thumbnail((800, 800))
+        img_tk = ImageTk.PhotoImage(img)
+        if 'img_label' not in img_container:
+            img_container['img_label'] = tk.Label(root)
+            img_container['img_label'].pack(side=tk.TOP, pady=10)
+        img_container['img_label'].configure(image=img_tk)
+        img_container['img_label'].image = img_tk
+    else:
+        messagebox.showerror("错误", "没有选择图片路径")
+
+def change_size(event, img, label_pic):
+    img_aspect = img.shape[1] / img.shape[0]
+    new_aspect = event.width / event.height
+
+    if new_aspect > img_aspect:
+        new_width = int(event.height * img_aspect)
+        new_height = event.height
+    else:
+        new_width = event.width
+        new_height = int(event.width / img_aspect)
+
+    resized_image = cv2.resize(img, (new_width, new_height))
+    image_tk = ImageTk.PhotoImage(Image.fromarray(resized_image))
+    label_pic.image = image_tk
+    label_pic['image'] = image_tk
+
+def savefile():
+    global edge
+
+    filename = filedialog.asksaveasfilename(defaultextension=".jpg", filetypes=[("JPEG files", "*.jpg"), ("PNG files", "*.png"), ("BMP files", "*.bmp")])
+    if not filename:
+        return
+
+    if edge is not None:
+        try:
+            edge.save(filename)
+            messagebox.showinfo("保存成功", "图片保存成功！")
+        except Exception as e:
+            messagebox.showerror("保存失败", f"无法保存图片: {e}")
+    else:
+        messagebox.showerror("保存失败", "没有图像可保存")
+
+def detect_and_show_processed_image(root, img_container):
+    global processed_win, edge
+
+    # 判断是否已经选取图片
+    if 'file_path' not in img_container or not img_container['file_path']:
+        messagebox.showerror("错误", "没有选择图片！")
+        return
+
+    file_path = img_container['file_path']
+    print(f"File path: {file_path}")  # 调试输出文件路径
+
+    # 调用图像处理函数
+    processed_img = locate_license_plate(file_path)
+
+    # 识别字符
+    recognized_text = recognize_characters(processed_img)
+
+    # 转换为RGB格式
+    processed_img_rgb = cv2.cvtColor(processed_img, cv2.COLOR_BGR2RGB)
+
+    # 更新 edge 变量
+    edge = Image.fromarray(processed_img_rgb)
+
+    # 创建Toplevel窗口
+    try:
+        processed_win.destroy()
+    except Exception:
+        print("NVM")
+    finally:
+        processed_win = Toplevel(root)
+        processed_win.attributes('-topmost', True)
+    processed_win.geometry("360x340")
+    processed_win.resizable(True, True)  # 可缩放
+    processed_win.title("处理后的图像")
+
+    # 显示图像
+    label_pic = tk.Label(processed_win, text="IMG", width=360, height=240)
+    image_tk = ImageTk.PhotoImage(Image.fromarray(processed_img_rgb))
+    label_pic.image = image_tk
+    label_pic['image'] = image_tk
+
+    label_pic.bind('<Configure>', lambda event: change_size(event, processed_img_rgb, label_pic))
+    label_pic.pack(fill=tk.BOTH, expand=tk.YES)
+
+    # 显示识别结果
+    result_label = tk.Label(processed_win, text=f"识别结果: {recognized_text}")
+    result_label.pack(side=tk.TOP, pady=10)
+
+    # 添加保存按钮
+    btn_save = tk.Button(processed_win, text="保存", bg='#add8e6', fg='black', font=('Helvetica', 14), width=20,
+                             command=savefile)
+    btn_save.pack(pady=10)
+
+    return