forked from BboyHanat/TextGenerator
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yml
100 lines (93 loc) · 2.77 KB
/
config.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
base:
# 启动的进程数
process_count: 2
# 每个进程下生成的图片个数
count_per_process: 5
# 是否生成voc数据
gen_voc: True
gen_lsvt: True
provider:
# 选取语料的配置
text:
- name: english
# 语料文本的路径
path: assets/corpus/word-en.txt
# 生成文本时，选取此语料的概率
probability: 0.3
# 读取语料的方式（逐行读取: line，逐字读取: word）
type: line
# 生成的文本长度区间
len_range: (2,20)
- name: chinese
path: assets/corpus/sentence.txt
probability: 0.3
type: line
len_range: (2,20)
- name: random_character
path: assets/corpus/chinese_synthetic.txt
probability: 0.1
type: word
len_range: (2,20)
# 候选区生成配置
smooth_area:
down_scale: 32
anchor_ratio: (0.17, 0.25, 0.5, 1.0, 2.0, 4.0, 6)
anchor_scale: (8, 16, 24, 32, 48, 64, 72, 90)
# 背景图选取配置
bg_img:
# 从磁盘选取
- type: from_dir
# 生成图片的背景图存放路径
dir: assets/img
# 选取磁盘上的背景图作为背景图的概率
probability: 0.8
# 自动生成
- type: from_generate
# 选取自动生成背景图作为背景图的概率
probability: 0.2
width_range: (1000, 1500)
height_range: (1000, 1500)
# 文字贴图配置
text_img:
# 文字文件存放路径
font_file_dir: assets/font
# 文字贴图输出路径
text_img_output_dir: output/text_img
# 文字贴图json信息输出路径
text_img_info_output_dir: output/text_img_info
# 字体最小尺寸
font_min_size: 15
# 字体最大尺寸
font_max_size: 50
# 使用指定颜色作为文本颜色的概率（相对于自动选取颜色）
use_char_common_color_probability: 0.5
# 指定字体的可选颜色列表（生成字体贴图时会从以下颜色列表中选取颜色作为字体颜色）
char_common_color_list:
- (0,0,0,255)
- (255,0,0,255)
# 文本框的宽度
char_border_width: 0
# 文本框的颜色
char_border_color: (255,255,0,255)
# 自动padding到指定的 长边/短边 的比例
auto_padding_to_ratio: 10
# 布局配置
layout:
# 输出布局文件的路径
out_put_dir: output
# 贴图旋转角度区间
rotate_angle_range: (-8, 8)
# 各种布局模式以及其选取的概率
strategy_list:
- name: HorizontalStrategy
probability: 0.2
- name: VerticalStrategy
probability: 0.2
- name: HorizontalFlowStrategy
probability: 0.2
- name: VerticalFlowStrategy
probability: 0.2
- name: RandomPasteStrategy
probability: 0.1
- name: CustomizationStrategy1
probability: 0.1