feat-特殊车牌生成器

chinese_license_plate_generator/plate_number.py

@@ -0,0 +1,112 @@
+"""
+generate plate numbers
+"""
+
+import numpy as np
+import cv2, os
+from glob import glob
+
+# 省份
+provinces = ["京", "津", "冀", "晋", "蒙", "辽", "吉", "黑", "沪",
+             "苏", "浙", "皖", "闽", "赣", "鲁", "豫", "鄂", "湘",
+             "粤", "桂", "琼", "渝", "川", "贵", "云", "藏", "陕",
+             "甘", "青", "宁", "新"]
+
+# "港", "澳", "使", "领", "学", "警", "挂"]
+digits = ['{}'.format(x + 1) for x in range(9)] + ['0']
+
+# 英文，没有I、O两个字符
+letters = [chr(x + ord('A')) for x in range(26) if not chr(x + ord('A')) in ['I', 'O']]
+# print('letters', digits + letters)
+
+# 随机选取
+def random_select(data):
+    return data[np.random.randint(len(data))]
+
+# 带权重的序号字符选取：数字概率远高于字母（真实车牌中字母占比约20%）
+def random_char():
+    if np.random.random() < 0.8:
+        return random_select(digits)
+    else:
+        return random_select(letters)
+
+# 蓝牌：省简称 + 发牌机关字母 + 序号（数字为主）
+def generate_plate_number_blue(length=7):
+    plate = random_select(provinces)
+    # 发牌机关代号必须是字母
+    plate += random_select(letters)
+    # 序号部分：数字80%，字母20%
+    for i in range(length - 2):
+        plate += random_char()
+    return plate
+
+# 黄色大型车（无后缀，普通黄牌）
+def generate_plate_number_yellow():
+    return generate_plate_number_blue()
+
+# 教练车
+def generate_plate_number_yellow_xue():
+    plate = generate_plate_number_blue()
+    return plate[:6] + '学'
+
+# 白色警车（末位必带"警"）
+def generate_plate_number_white():
+    plate = generate_plate_number_blue()
+    return plate[:6] + '警'
+
+# 军车/武警首位字母（2013/2019式，非随机）
+army_letters = ['V',  # 中央军委
+                'L',  # 陆军
+                'H',  # 海军
+                'K',  # 空军
+                'J',  # 火箭军
+                'Z',  # 战略支援部队
+                'B',  # 联勤保障部队
+                'W']  # 武警
+
+# 白色军车/武警（首位从军车字母中选取，不带"警"）
+def generate_plate_number_army():
+    first_letter = random_select(army_letters)
+    plate = generate_plate_number_blue()
+    return first_letter + plate[1:]
+
+
+def generate_plate_number_black_gangao():
+    plate = generate_plate_number_blue()
+    return '粤' + plate[1:6] + random_select(["港", "澳"])
+
+
+# ---- 使领馆旧版（≤2017.1.9） ----
+# 使馆旧版：使XXX·XXX（"使"在首位）
+def generate_plate_number_black_shi_old():
+    plate = generate_plate_number_blue()
+    return '使' + plate[1:]
+
+# 领馆旧版：省A·XXXX领（带城市字母，"领"在末位）
+def generate_plate_number_black_ling_old():
+    plate = generate_plate_number_blue()
+    return plate[:6] + '领'
+
+# ---- 使领馆新版（2017.1.9~） ----
+# 使馆新版：XXX·XXX使（3位国代码 + 3位流水 + "使"，白字）
+def generate_plate_number_black_shi_new():
+    code = '{:03d}'.format(np.random.randint(101, 281))
+    serial = '{:03d}'.format(np.random.randint(1, 1000))
+    return code + serial + '使'
+
+# 领馆新版：省XXX·XX领（省 + 3位国代码 + 2位流水 + "领"，白字）
+def generate_plate_number_black_ling_new():
+    province = random_select(provinces)
+    code = '{:03d}'.format(np.random.randint(101, 281))
+    serial = '{:02d}'.format(np.random.randint(1, 100))
+    return province + code + serial + '领'
+
+
+def board_bbox(polys):
+    x1, y1 = np.min(polys, axis=0)
+    x2, y2 = np.max(polys, axis=0)
+
+    return [x1, y1, x2, y2]
+
+if __name__ == '__main__':
+    pass