def __init__(self,
corpus,
fonts,
bgs,
cfg,
width=256,
height=32,
clip_max_chars=False,
debug=False,
gpu=False,
strict=False):
self.corpus = corpus
self.fonts = fonts
self.bgs = bgs
self.out_width = width
self.out_height = height
self.clip_max_chars = clip_max_chars
self.max_chars = math.floor(width / 4) - 1
self.debug = debug
self.gpu = gpu
self.strict = strict
self.cfg = cfg
self.timer = Timer()
self.liner = Liner(cfg)
self.noiser = Noiser(cfg)
self.remaper = Remaper(cfg)
self.create_kernals()
if self.strict:
self.font_unsupport_chars = font_utils.get_unsupported_chars(
self.fonts, corpus.chars_file)
def __init__(self,
corpus,
fonts,
bgs,
cfg,
width=256,
height=32,
clip_max_chars=False,
debug=True,
gpu=False,
strict=False):
self.cnt = 0
self.corpus = corpus
self.fonts = fonts
self.bgs = bgs
self.out_width = width
self.out_height = height
self.clip_max_chars = clip_max_chars
self.max_chars = math.floor(width / 4) - 1
self.debug = debug
self.gpu = gpu
self.strict = strict
self.cfg = cfg
self.timer = Timer()
self.liner = Liner(cfg)
self.noiser = Noiser(cfg)
self.remaper = Remaper(cfg)
self.create_kernals()
if not self.is_bgr():
for i, bg in enumerate(self.bgs):
self.bgs[i] = cv2.cvtColor(bg, cv2.COLOR_BGR2GRAY)
if self.strict:
self.font_unsupport_chars = font_utils.get_unsupported_chars(
self.fonts, corpus.chars_file)
from textrenderer.renderer import Renderer
from tenacity import retry
lock = mp.Lock()
counter = mp.Value('i', 0)
STOP_TOKEN = 'kill'
flags = parse_args()
cfg = load_config(flags.config_file)
fonts = font_utils.get_font_paths_from_list(flags.fonts_list)
bgs = utils.load_bgs(flags.bg_dir)
corpus = corpus_factory(flags.corpus_mode, flags.chars_file, flags.corpus_dir,
flags.length)
font_unsupport_chars = font_utils.get_unsupported_chars(
fonts, corpus.chars_file)
def start_listen(q, fname):
""" listens for messages on the q, writes to file. """
f = open(fname, mode='a', encoding='utf-8')
while 1:
m = q.get()
if m == STOP_TOKEN:
break
try:
f.write(str(m) + '\n')
except:
traceback.print_exc()
