第一步:准备训练样本
样本比较单一,采用电脑自带可用字体,经筛选,大概只有不到30个。通过调节大小,使其尽量丰富.
# encoding = utf-8
import numpy as np
import os
import cv2
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
import time
import random
# 创建汉字背景图片
def backgroundPic():
img = np.ones((600, 600), dtype=np.uint8)
cv2.imwrite(r'background.jpg', img)
img = cv2.imread(r'o.jpg')
ret, img = cv2.threshold(img, 0, 255, cv2.THRESH_BINARY)
cv2.imwrite(r'background.jpg', img)
img = Image.open('o.jpg')
return img
def writeChineseWord(chineseWord, directory):
img = backgroundPic()
# 可用字体列表
usable_font = ['SIMYOU.TTF', 'simsun.ttc', 'simsun.ttc', 'SIMLI.TTF', 'simkai.ttf', 'simhei.ttf', 'STCAIYUN.TTF',
'STFANGSO.TTF', 'STXINGKA.TTF', 'STHUPO.TTF', 'STKAITI.TTF', 'STHUPO.TTF', 'STKAITI.TTF',
'STHUPO.TTF', 'STKAITI.TTF', 'STLITI.TTF', 'STSONG.TTF', 'STXIHEI.TTF', 'STXINWEI.TTF',
'STZHONGS.TTF', 'simfang.ttf', 'FZYTK.TTF', 'FZSTK.TTF', 'AliHYAiHei.ttf',
'msgothic.ttc', 'msgothic.ttc', 'msgothic