29 lines
848 B
Python
29 lines
848 B
Python
#!/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)')
|