diff --git a/util/general_utils.py b/util/general_utils.py index 0524d0655..c00016368 100644 --- a/util/general_utils.py +++ b/util/general_utils.py @@ -105,15 +105,21 @@ def word2img(title: str, text: str, max_width=30, font_size=20): raise Exception("找不到字体文件") width_factor = 1.0 - height_factor = 1.8 + height_factor = 1.5 # 格式化文本宽度最大为30 lines = text.split('\n') i = 0 length = len(lines) for l in lines: if len(l) > max_width: - lines[i] = l[:max_width] + '\n' + l[max_width:] - length += 1 + # lines[i] = l[:max_width] + '\n' + l[max_width:] + # for + cp = l + for ii in range(len(l)): + if ii % max_width == 0: + cp = cp[:ii] + '\n' + cp[ii:] + length += 1 + lines[i] = cp i += 1 text = '\n'.join(lines) width = int(max_width * font_size * width_factor)