Пример #1
0
from parse_args import parse_args
import libs.utils as utils
import libs.font_utils as font_utils
from textrenderer.corpus.corpus_utils import corpus_factory
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)

renderer = Renderer(corpus,
                    fonts,
                    bgs,
                    cfg,
                    height=flags.img_height,
                    width=flags.img_width,
                    clip_max_chars=flags.clip_max_chars,
                    debug=flags.debug,
                    gpu=flags.gpu,
                    strict=flags.strict)
Пример #2
0
import libs.utils as utils
import libs.font_utils as font_utils
from textrenderer.corpus.corpus_utils import corpus_factory
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)  # fonts-字体文件路径组成的列表-目前的文件就1个路径
bgs = utils.load_bgs(flags.bg_dir)  # 背景图像组成的列表

corpus = corpus_factory(flags.corpus_mode, flags.chars_file, flags.corpus_dir,
                        flags.length)  # 导入语料库-类方法

renderer = Renderer(corpus,
                    fonts,
                    bgs,
                    cfg,
                    height=flags.img_height,
                    width=flags.img_width,
                    clip_max_chars=flags.clip_max_chars,
                    debug=flags.debug,
                    gpu=flags.gpu,
                    strict=flags.strict)  # 渲染器-总方法