狗操的大作业
This commit is contained in:
@@ -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()
|
||||
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 2.6 MiB |
Binary file not shown.
|
After Width: | Height: | Size: 172 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 455 KiB |
@@ -0,0 +1,20 @@
|
||||
{
|
||||
"config":[
|
||||
{
|
||||
"open":1,
|
||||
"blur":3,
|
||||
"morphologyr":4,
|
||||
"morphologyc":19,
|
||||
"col_num_limit":10,
|
||||
"row_num_limit":21
|
||||
},
|
||||
{
|
||||
"open":0,
|
||||
"blur":3,
|
||||
"morphologyr":5,
|
||||
"morphologyc":12,
|
||||
"col_num_limit":10,
|
||||
"row_num_limit":18
|
||||
}
|
||||
]
|
||||
}
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 4.0 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 251 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 270 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 288 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 158 KiB |
@@ -0,0 +1,402 @@
|
||||
label: 0
|
||||
display_name: "background"
|
||||
label: 1
|
||||
display_name: "accordion"
|
||||
label: 2
|
||||
display_name: "airplane"
|
||||
label: 3
|
||||
display_name: "ant"
|
||||
label: 4
|
||||
display_name: "antelope"
|
||||
label: 5
|
||||
display_name: "apple"
|
||||
label: 6
|
||||
display_name: "armadillo"
|
||||
label: 7
|
||||
display_name: "artichoke"
|
||||
label: 8
|
||||
display_name: "axe"
|
||||
label: 9
|
||||
display_name: "baby_bed"
|
||||
label: 10
|
||||
display_name: "backpack"
|
||||
label: 11
|
||||
display_name: "bagel"
|
||||
label: 12
|
||||
display_name: "balance_beam"
|
||||
label: 13
|
||||
display_name: "banana"
|
||||
label: 14
|
||||
display_name: "band_aid"
|
||||
label: 15
|
||||
display_name: "banjo"
|
||||
label: 16
|
||||
display_name: "baseball"
|
||||
label: 17
|
||||
display_name: "basketball"
|
||||
label: 18
|
||||
display_name: "bathing_cap"
|
||||
label: 19
|
||||
display_name: "beaker"
|
||||
label: 20
|
||||
display_name: "bear"
|
||||
label: 21
|
||||
display_name: "bee"
|
||||
label: 22
|
||||
display_name: "bell_pepper"
|
||||
label: 23
|
||||
display_name: "bench"
|
||||
label: 24
|
||||
display_name: "bicycle"
|
||||
label: 25
|
||||
display_name: "binder"
|
||||
label: 26
|
||||
display_name: "bird"
|
||||
label: 27
|
||||
display_name: "bookshelf"
|
||||
label: 28
|
||||
display_name: "bow_tie"
|
||||
label: 29
|
||||
display_name: "bow"
|
||||
label: 30
|
||||
display_name: "bowl"
|
||||
label: 31
|
||||
display_name: "brassiere"
|
||||
label: 32
|
||||
display_name: "burrito"
|
||||
label: 33
|
||||
display_name: "bus"
|
||||
label: 34
|
||||
display_name: "butterfly"
|
||||
label: 35
|
||||
display_name: "camel"
|
||||
label: 36
|
||||
display_name: "can_opener"
|
||||
label: 37
|
||||
display_name: "car"
|
||||
label: 38
|
||||
display_name: "cart"
|
||||
label: 39
|
||||
display_name: "cattle"
|
||||
label: 40
|
||||
display_name: "cello"
|
||||
label: 41
|
||||
display_name: "centipede"
|
||||
label: 42
|
||||
display_name: "chain_saw"
|
||||
label: 43
|
||||
display_name: "chair"
|
||||
label: 44
|
||||
display_name: "chime"
|
||||
label: 45
|
||||
display_name: "cocktail_shaker"
|
||||
label: 46
|
||||
display_name: "coffee_maker"
|
||||
label: 47
|
||||
display_name: "computer_keyboard"
|
||||
label: 48
|
||||
display_name: "computer_mouse"
|
||||
label: 49
|
||||
display_name: "corkscrew"
|
||||
label: 50
|
||||
display_name: "cream"
|
||||
label: 51
|
||||
display_name: "croquet_ball"
|
||||
label: 52
|
||||
display_name: "crutch"
|
||||
label: 53
|
||||
display_name: "cucumber"
|
||||
label: 54
|
||||
display_name: "cup_or_mug"
|
||||
label: 55
|
||||
display_name: "diaper"
|
||||
label: 56
|
||||
display_name: "digital_clock"
|
||||
label: 57
|
||||
display_name: "dishwasher"
|
||||
label: 58
|
||||
display_name: "dog"
|
||||
label: 59
|
||||
display_name: "domestic_cat"
|
||||
label: 60
|
||||
display_name: "dragonfly"
|
||||
label: 61
|
||||
display_name: "drum"
|
||||
label: 62
|
||||
display_name: "dumbbell"
|
||||
label: 63
|
||||
display_name: "electric_fan"
|
||||
label: 64
|
||||
display_name: "elephant"
|
||||
label: 65
|
||||
display_name: "face_powder"
|
||||
label: 66
|
||||
display_name: "fig"
|
||||
label: 67
|
||||
display_name: "filing_cabinet"
|
||||
label: 68
|
||||
display_name: "flower_
|
||||
label: 69
|
||||
display_name: "fl
|
||||
label: 70
|
||||
display_name: "fox"
|
||||
label: 71
|
||||
display_name: "french_h
|
||||
label: 72
|
||||
display_name: "frog"
|
||||
label: 73
|
||||
display_name: "frying_
|
||||
label: 74
|
||||
display_name: "giant_pa
|
||||
label: 75
|
||||
display_name: "goldfish"
|
||||
label: 76
|
||||
display_name: "golf_ball"
|
||||
label: 77
|
||||
display_name: "golfcart"
|
||||
label: 78
|
||||
display_name: "guacamole"
|
||||
label: 79
|
||||
display_name: "guitar"
|
||||
label: 80
|
||||
display_name: "hair_dryer"
|
||||
label: 81
|
||||
display_name: "hair_spray"
|
||||
label: 82
|
||||
display_name: "hamburger"
|
||||
label: 83
|
||||
display_name: "hammer"
|
||||
label: 84
|
||||
display_name: "hamster"
|
||||
label: 85
|
||||
display_name: "harmonica"
|
||||
label: 86
|
||||
display_name: "harp"
|
||||
label: 87
|
||||
display_name: "hat_with_a_wide_brim"
|
||||
label: 88
|
||||
display_name: "head_cabbage"
|
||||
label: 89
|
||||
display_name: "helmet"
|
||||
label: 90
|
||||
display_name: "hippopotamus"
|
||||
label: 91
|
||||
display_name: "horizontal_bar"
|
||||
label: 92
|
||||
display_name: "horse"
|
||||
label: 93
|
||||
display_name: "hotdog"
|
||||
label: 94
|
||||
display_name: "iPod"
|
||||
label: 95
|
||||
display_name: "isopod"
|
||||
label: 96
|
||||
display_name: "jellyfish"
|
||||
label: 97
|
||||
display_name: "koala_bear"
|
||||
label: 98
|
||||
display_name: "ladle"
|
||||
label: 99
|
||||
display_name: "ladybug"
|
||||
label: 100
|
||||
display_name: "lamp"
|
||||
label: 101
|
||||
display_name: "laptop"
|
||||
label: 102
|
||||
display_name: "lemon"
|
||||
label: 103
|
||||
display_name: "lion"
|
||||
label: 104
|
||||
display_name: "lipstick"
|
||||
label: 105
|
||||
display_name: "lizard"
|
||||
label: 106
|
||||
display_name: "lobster"
|
||||
label: 107
|
||||
display_name: "maillot"
|
||||
label: 108
|
||||
display_name: "maraca"
|
||||
label: 109
|
||||
display_name: "microphone"
|
||||
label: 110
|
||||
display_name: "microwave"
|
||||
label: 111
|
||||
display_name: "milk_can"
|
||||
label: 112
|
||||
display_name: "miniskirt"
|
||||
label: 113
|
||||
display_name: "monkey"
|
||||
label: 114
|
||||
display_name: "motorcycle"
|
||||
label: 115
|
||||
display_name: "mushroom"
|
||||
label: 116
|
||||
display_name: "nail"
|
||||
label: 117
|
||||
display_name: "neck_brace"
|
||||
label: 118
|
||||
display_name: "oboe"
|
||||
label: 119
|
||||
display_name: "orange"
|
||||
label: 120
|
||||
display_name: "otter"
|
||||
label: 121
|
||||
display_name: "pencil_box"
|
||||
label: 122
|
||||
display_name: "pencil_sharpener"
|
||||
label: 123
|
||||
display_name: "perfume"
|
||||
label: 124
|
||||
display_name: "person"
|
||||
label: 125
|
||||
display_name: "piano"
|
||||
label: 126
|
||||
display_name: "pineapple"
|
||||
label: 127
|
||||
display_name: "ping-pong_ball"
|
||||
label: 128
|
||||
display_name: "pitcher"
|
||||
label: 129
|
||||
display_name: "pizza"
|
||||
label: 130
|
||||
display_name: "plastic_bag"
|
||||
label: 131
|
||||
display_name: "plate_rack"
|
||||
label: 132
|
||||
display_name: "pomegranate"
|
||||
label: 133
|
||||
display_name: "popsicle"
|
||||
label: 134
|
||||
display_name: "porcupine"
|
||||
label: 135
|
||||
display_name: "power_drill"
|
||||
label: 136
|
||||
display_name: "pretzel"
|
||||
label: 137
|
||||
display_name: "printer"
|
||||
label: 138
|
||||
display_name: "puck"
|
||||
label: 139
|
||||
display_name: "punching_bag"
|
||||
label: 140
|
||||
display_name: "purse"
|
||||
label: 141
|
||||
display_name: "rabbit"
|
||||
label: 142
|
||||
display_name: "racket"
|
||||
label: 143
|
||||
display_name: "ray"
|
||||
label: 144
|
||||
display_name: "red_panda"
|
||||
label: 145
|
||||
display_name: "refrigerator"
|
||||
label: 146
|
||||
display_name: "remote_control"
|
||||
label: 147
|
||||
display_name: "rubber_eraser"
|
||||
label: 148
|
||||
display_name: "rugby_ball"
|
||||
label: 149
|
||||
display_name: "ruler"
|
||||
label: 150
|
||||
display_name: "salt_or_pepper_shaker"
|
||||
label: 151
|
||||
display_name: "saxophone"
|
||||
label: 152
|
||||
display_name: "scorpion"
|
||||
label: 153
|
||||
display_name: "screwdriver"
|
||||
label: 154
|
||||
display_name: "seal"
|
||||
label: 155
|
||||
display_name: "sheep"
|
||||
label: 156
|
||||
display_name: "ski"
|
||||
label: 157
|
||||
display_name: "skunk"
|
||||
label: 158
|
||||
display_name: "snail"
|
||||
label: 159
|
||||
display_name: "snake"
|
||||
label: 160
|
||||
display_name: "snowmobile"
|
||||
label: 161
|
||||
display_name: "snowplow"
|
||||
label: 162
|
||||
display_name: "soap_dispenser"
|
||||
label: 163
|
||||
display_name: "soccer_ball"
|
||||
label: 164
|
||||
display_name: "sofa"
|
||||
label: 165
|
||||
display_name: "spatula"
|
||||
label: 166
|
||||
display_name: "squirrel"
|
||||
label: 167
|
||||
display_name: "starfish"
|
||||
label: 168
|
||||
display_name: "stethoscope"
|
||||
label: 169
|
||||
display_name: "stove"
|
||||
label: 170
|
||||
display_name: "strainer"
|
||||
label: 171
|
||||
display_name: "strawberry"
|
||||
label: 172
|
||||
display_name: "stretcher"
|
||||
label: 173
|
||||
display_name: "sunglasses"
|
||||
label: 174
|
||||
display_name: "swimming_trunks"
|
||||
label: 175
|
||||
display_name: "swine"
|
||||
label: 176
|
||||
display_name: "syringe"
|
||||
label: 177
|
||||
display_name: "table"
|
||||
label: 178
|
||||
display_name: "tape_player"
|
||||
label: 179
|
||||
display_name: "tennis_ball"
|
||||
label: 180
|
||||
display_name: "tick"
|
||||
label: 181
|
||||
display_name: "tie"
|
||||
label: 182
|
||||
display_name: "tiger"
|
||||
label: 183
|
||||
display_name: "toaster"
|
||||
label: 184
|
||||
display_name: "traffic_light"
|
||||
label: 185
|
||||
display_name: "train"
|
||||
label: 186
|
||||
display_name: "trombone"
|
||||
label: 187
|
||||
display_name: "trumpet"
|
||||
label: 188
|
||||
display_name: "turtle"
|
||||
label: 189
|
||||
display_name: "tv_or_monitor"
|
||||
label: 190
|
||||
display_name: "unicycle"
|
||||
label: 191
|
||||
display_name: "vacuum"
|
||||
label: 192
|
||||
display_name: "violin"
|
||||
label: 193
|
||||
display_name: "volleyball"
|
||||
label: 194
|
||||
display_name: "waffle_iron"
|
||||
label: 195
|
||||
display_name: "washer"
|
||||
label: 196
|
||||
display_name: "water_bottle"
|
||||
label: 197
|
||||
display_name: "watercraft"
|
||||
label: 198
|
||||
display_name: "whale"
|
||||
label: 199
|
||||
display_name: "wine_bottle"
|
||||
label: 200
|
||||
display_name: "zebra"
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 42 KiB |
Reference in New Issue
Block a user