-
Notifications
You must be signed in to change notification settings - Fork 51
/
Copy pathlicense_plate_elements.py
170 lines (148 loc) · 6.81 KB
/
license_plate_elements.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
# -*- coding: utf-8 -*-
"""
Created on 2019/4/17
File license_plate_elements
@author:ZhengYuwei
功能：
定义车牌相关的元素，包括车牌中的字符和车牌类型
一般车牌字符一共7位，新能源车牌8位，故增加一个标志位，表示车牌位数
plate_type_enum为车牌类型，is_exist_plate_enum标记是否存在车牌
"""
class LicensePlateElements(object):
""" 定义车牌相关的元素（车牌字符和车牌类型），以及获取元素的方法
一般车牌字符一共7位，新能源车牌8位，故增加一个标志位，表示车牌位数
plate_type_enum为车牌类型，is_exist_plate_enum标记是否存在车牌
"""
# 车牌第1位字符的取值范围及其label
char1_enum = {
u"京": 0, u"沪": 1, u"津": 2, u"渝": 3, u"冀": 4, u"晋": 5, u"蒙": 6, u"辽": 7, u"吉": 8, u"黑": 9,
u"苏": 10, u"浙": 11, u"皖": 12, u"闽": 13, u"赣": 14, u"鲁": 15, u"豫": 16, u"鄂": 17, u"湘": 18, u"粤": 19,
u"桂": 20, u"琼": 21, u"川": 22, u"贵": 23, u"云": 24, u"藏": 25, u"陕": 26, u"甘": 27, u"青": 28, u"宁": 29,
u"新": 30, u"军": 31, u"使": 32,
}
# 车牌第2位字符的取值范围及其label
char2_enum = {
"A": 0, "B": 1, "C": 2, "D": 3, "E": 4, "F": 5, "G": 6, "H": 7, "J": 8, "K": 9,
"L": 10, "M": 11, "N": 12, "P": 13, "Q": 14, "R": 15, "S": 16, "T": 17, "U": 18, "V": 19,
"W": 20, "X": 21, "Y": 22, "Z": 23, "0": 24, "1": 25, "2": 26, "3": 27, "4": 28, "5": 29,
"6": 30, "7": 31, "8": 32, "9": 33,
}
# 车牌第3~6位字符的取值范围及其label
char3_6_enum = {
"0": 0, "1": 1, "2": 2, "3": 3, "4": 4, "5": 5, "6": 6, "7": 7, "8": 8, "9": 9,
"A": 10, "B": 11, "C": 12, "D": 13, "E": 14, "F": 15, "G": 16, "H": 17, "J": 18, "K": 19,
"L": 20, "M": 21, "N": 22, "P": 23, "Q": 24, "R": 25, "S": 26, "T": 27, "U": 28, "V": 29,
"W": 30, "X": 31, "Y": 32, "Z": 33,
}
# 车牌第7位字符的取值范围及其label
char7_enum = {
"0": 0, "1": 1, "2": 2, "3": 3, "4": 4, "5": 5, "6": 6, "7": 7, "8": 8, "9": 9,
"A": 10, "B": 11, "C": 12, "D": 13, "E": 14, "F": 15, "G": 16, "H": 17, "J": 18, "K": 19,
"L": 20, "M": 21, "N": 22, "P": 23, "Q": 24, "R": 25, "S": 26, "T": 27, "U": 28, "V": 29,
"W": 30, "X": 31, "Y": 32, "Z": 33, u"学": 34, u"警": 35, u"领": 36, u"挂": 37, u"港": 38, u"澳": 39,
u"使": 40,
}
# 车牌第8位字符的取值范围及其label
char8_enum = {
"0": 0, "1": 1, "2": 2, "3": 3, "4": 4, "5": 5, "6": 6, "7": 7, "8": 8, "9": 9, "D": 10, "F": 11,
}
# 不同车牌位置与车牌字符变量取值表的映射表
index_to_char_label = {
0: char1_enum,
1: char2_enum,
2: char3_6_enum,
3: char3_6_enum,
4: char3_6_enum,
5: char3_6_enum,
6: char7_enum,
7: char8_enum,
}
# 车牌位数及其label
char_number_enum = {'7': 0, '8': 1}
# 车牌类型
plate_type_enum = {
'single_blue': 0, 'single_yellow': 1, 'double_yellow': 2, 'police': 3, 'learner': 4,
'hk': 5, 'macau': 6, 'single_army': 7, 'double_army': 8, 'consulate': 9,
'embassy': 10, 'army_police': 11, 'double_army_police': 12, 'small_new_energy': 13, 'big_new_energy': 14,
'factory': 15, 'single_black': 16, 'other': 17,
}
# 车牌颜色
plate_color_indices = {'blue': 0, 'yellow': 1, 'green': 2, 'white': 3, 'black': 4, 'other': 5, }
# 是否存在车牌
is_exist_plate_enum = {'exist': 0, 'non-exist': 1}
def __init__(self):
""" 初始化，通过每一位车牌字符的char_label，增加label_char的反向映射 """
self.index_to_label_char = {}
for index, char_label in LicensePlateElements.index_to_char_label.items():
self.index_to_label_char[index] = {label: char for char, label in char_label.items()}
self.label_to_type = {}
for label, plate_type in LicensePlateElements.plate_type_enum.items():
self.label_to_type[label] = plate_type
return
def get_char_label(self, index):
""" 获取指定车牌位置的char-label映射
:param index: 指定车牌位数
:return: 指定车牌位置的char-label映射
"""
return self.index_to_char_label.get(index, None)
def get_label_char(self, index):
""" 获取指定车牌位置的label-char映射
:param index: 指定车牌位数
:return: 指定车牌位置的label-char映射
"""
return self.index_to_label_char.get(index, None)
def get_chars_sorted_by_label(self, index):
""" 获取指定车牌位置上，按label排序的字符列表
:param index: 指定车牌位置
:return: 字符列表
"""
label_char = self.get_label_char(index)
if label_char is None:
print('Index is Out of Maximal License Plate Figures')
chars = [None] * len(label_char)
for index, char in label_char.items():
chars[index] = char
return chars
def get_char(self, index, label):
""" 获取指定车牌位数、标签的对应字符
:param index: 指定车牌位数
:param label: 指定标签
:return: 车牌字符
"""
label_char = self.index_to_label_char.get(index, None)
if label_char is None:
print('Index is Out of Maximal License Plate Figures')
return None
return label_char.get(label, None)
def get_label(self, index, char):
""" 获取指定车牌位数、字符的对应标签
:param index: 指定车牌位置
:param char: 指定字符
:return: 车牌标签
"""
char_label = self.index_to_char_label.get(index, None)
if char_label is None:
print('Index is Out of Maximal License Plate Figures')
return None
return char_label.get(char, None)
def convert_to_labels(self, license_plate):
""" 将车牌号转换为标签
:param license_plate: 车牌号
:return: 标签
"""
labels = [None] * len(license_plate)
for index, char in enumerate(license_plate):
labels[index] = self.get_label(index, char)
return labels
def get_type_list(self):
""" 获取车牌类型有序（label序）列表 """
type_list = [None] * len(self.plate_type_enum)
for plate_type, label in self.plate_type_enum.items():
type_list[label] = plate_type
return type_list
def get_type(self, label):
""" 根据车配类型的label，获取车牌类型
:param label: 车牌类型的label
:return:
"""
return self.label_to_type[label]