特殊车牌数据集构建

tools/gen_pure_plates.py

@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+"""生成纯合成车牌（无场景背景）作为 CUT Domain A"""
+import sys, os
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'chinese_license_plate_generator'))
+from generate_multi_plate import MultiPlateGenerator
+import cv2, random
+import numpy as np
+
+random.seed(42)
+np.random.seed(42)
+
+g = MultiPlateGenerator(
+    'chinese_license_plate_generator/plate_model',
+    'chinese_license_plate_generator/font_model'
+)
+
+out = 'gan_data/domain_A_synthetic/police_pure'
+os.makedirs(out, exist_ok=True)
+
+for i in range(500):
+    num, bg, _ = g.generate_plate_number()
+    # 强制police类型
+    if '警' not in num:
+        num = num[:6] + '警'
+    img = g.generate_plate_special(num, 'white', False)
+    cv2.imwrite(f'{out}/synth_{i:04d}.jpg', img)
+
+print(f'Done: {out} ({len(os.listdir(out))} images)')