狗操的大作业

2024-12-18 18:38:45 +08:00
parent 7d0d76acbb
commit 99dba333c3
44 changed files with 1327 additions and 0 deletions
@@ -0,0 +1,154 @@
+import tkinter as tk
+from tkinter.filedialog import *
+from tkinter import ttk
+import tkinter.messagebox as mBox
+
+import predict
+import cv2
+from PIL import Image, ImageTk
+import threading
+import time
+
+
+
+class Surface(ttk.Frame):
+	pic_path = ""
+	viewhigh = 600
+	viewwide = 600
+	update_time = 0
+	thread = None
+	thread_run = False
+	camera = None
+	color_transform = {"green":("绿牌","#55FF55"), "yello":("黄牌","#FFFF00"), "blue":("蓝牌","#6666FF")}
+		
+	def __init__(self, win):
+		ttk.Frame.__init__(self, win)
+		frame_left = ttk.Frame(self)
+		frame_right1 = ttk.Frame(self)
+		frame_right2 = ttk.Frame(self)
+		win.title("车牌识别")
+		win.state("zoomed")
+		self.pack(fill=tk.BOTH, expand=tk.YES, padx="5", pady="5")
+		frame_left.pack(side=tk.LEFT,expand=1,fill=tk.BOTH)
+		frame_right1.pack(side=tk.TOP,expand=1,fill=tk.Y)
+		frame_right2.pack(side=tk.RIGHT,expand=0)
+		ttk.Label(frame_left, text='原图：').pack(anchor="nw") 
+		ttk.Label(frame_right1, text='车牌位置：').grid(column=0, row=0, sticky=tk.W)
+		
+		from_pic_ctl = ttk.Button(frame_right2, text="来自图片", width=20, command=self.from_pic)
+		from_vedio_ctl = ttk.Button(frame_right2, text="来自摄像头", width=20, command=self.from_vedio)
+		self.image_ctl = ttk.Label(frame_left)
+		self.image_ctl.pack(anchor="nw")
+		
+		self.roi_ctl = ttk.Label(frame_right1)
+		self.roi_ctl.grid(column=0, row=1, sticky=tk.W)
+		ttk.Label(frame_right1, text='识别结果：').grid(column=0, row=2, sticky=tk.W)
+		self.r_ctl = ttk.Label(frame_right1, text="")
+		self.r_ctl.grid(column=0, row=3, sticky=tk.W)
+		self.color_ctl = ttk.Label(frame_right1, text="", width="20")
+		self.color_ctl.grid(column=0, row=4, sticky=tk.W)
+		from_vedio_ctl.pack(anchor="se", pady="5")
+		from_pic_ctl.pack(anchor="se", pady="5")
+		self.predictor = predict.CardPredictor()
+		self.predictor.train_svm()
+		
+	def get_imgtk(self, img_bgr):
+		img = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB)
+		im = Image.fromarray(img)
+		imgtk = ImageTk.PhotoImage(image=im)
+		wide = imgtk.width()
+		high = imgtk.height()
+		if wide > self.viewwide or high > self.viewhigh:
+			wide_factor = self.viewwide / wide
+			high_factor = self.viewhigh / high
+			factor = min(wide_factor, high_factor)
+			
+			wide = int(wide * factor)
+			if wide <= 0 : wide = 1
+			high = int(high * factor)
+			if high <= 0 : high = 1
+			im=im.resize((wide, high), Image.LANCZOS) #在pillow的10.0.0版本中，ANTIALIAS方法被删除了，使用新的方法即可：
+			imgtk = ImageTk.PhotoImage(image=im)
+		return imgtk
+	
+	def show_roi(self, r, roi, color):
+		if r :
+			roi = cv2.cvtColor(roi, cv2.COLOR_BGR2RGB)
+			roi = Image.fromarray(roi)
+			self.imgtk_roi = ImageTk.PhotoImage(image=roi)
+			self.roi_ctl.configure(image=self.imgtk_roi, state='enable')
+			self.r_ctl.configure(text=str(r))
+			self.update_time = time.time()
+			try:
+				c = self.color_transform[color]
+				self.color_ctl.configure(text=c[0], background=c[1], state='enable')
+			except: 
+				self.color_ctl.configure(state='disabled')
+		elif self.update_time + 8 < time.time():
+			self.roi_ctl.configure(state='disabled')
+			self.r_ctl.configure(text="")
+			self.color_ctl.configure(state='disabled')
+		
+	def from_vedio(self):
+		if self.thread_run:
+			return
+		if self.camera is None:
+			self.camera = cv2.VideoCapture(0)
+			if not self.camera.isOpened():
+				mBox.showwarning('警告', '摄像头打开失败！')
+				self.camera = None
+				return
+		self.thread = threading.Thread(target=self.vedio_thread, args=(self,))
+		self.thread.setDaemon(True)
+		self.thread.start()
+		self.thread_run = True
+		
+	def from_pic(self):
+		self.thread_run = False
+		self.pic_path = askopenfilename(title="选择识别图片", filetypes=[("jpg图片", "*.jpg")])
+		if self.pic_path:
+			img_bgr = predict.imreadex(self.pic_path)
+			self.imgtk = self.get_imgtk(img_bgr)
+			self.image_ctl.configure(image=self.imgtk)
+			resize_rates = (1, 0.9, 0.8, 0.7, 0.6, 0.5, 0.4)
+			for resize_rate in resize_rates:
+				print("resize_rate:", resize_rate)
+				try:
+					r, roi, color = self.predictor.predict(img_bgr, resize_rate)
+				except:
+					continue
+				if r:
+					break
+			#r, roi, color = self.predictor.predict(img_bgr, 1)
+			self.show_roi(r, roi, color)
+
+	@staticmethod
+	def vedio_thread(self):
+		self.thread_run = True
+		predict_time = time.time()
+		while self.thread_run:
+			_, img_bgr = self.camera.read()
+			self.imgtk = self.get_imgtk(img_bgr)
+			self.image_ctl.configure(image=self.imgtk)
+			if time.time() - predict_time > 2:
+				r, roi, color = self.predictor.predict(img_bgr)
+				self.show_roi(r, roi, color)
+				predict_time = time.time()
+		print("run end")
+		
+		
+def close_window():
+	print("destroy")
+	if surface.thread_run :
+		surface.thread_run = False
+		surface.thread.join(2.0)
+	win.destroy()
+	
+	
+if __name__ == '__main__':
+	win=tk.Tk()
+	
+	surface = Surface(win)
+	win.protocol('WM_DELETE_WINDOW', close_window)
+	win.mainloop()
+