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()