From 9f55d17f6ab78bb95beee61ac9495bd7b469c0a4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=B4=E7=83=9C?= Date: Sun, 24 Nov 2024 16:48:17 -0800 Subject: [PATCH 1/5] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E9=87=8D=E6=9E=84:=20?= =?UTF-8?q?=E5=91=BD=E5=90=8D,=20=E5=8E=98=E6=B8=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 2 +- img2txt.py | 50 +++++++++++++++++++++++++------------------------- 2 files changed, 26 insertions(+), 26 deletions(-) diff --git a/README.md b/README.md index 8c8a1df..9fe657e 100644 --- a/README.md +++ b/README.md @@ -1,7 +1,7 @@ # 图像字符化 -## 介绍 +## 功能介绍 ## 图片转换为文本 diff --git a/img2txt.py b/img2txt.py index 44bda99..0e53bcf 100644 --- a/img2txt.py +++ b/img2txt.py @@ -20,33 +20,33 @@ def get_args(): def main(opt): if opt.mode == "simple": - CHAR_LIST = '@%#*+=-:. ' + 字符列表 = '@%#*+=-:. ' else: - CHAR_LIST = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. " - num_chars = len(CHAR_LIST) - num_cols = opt.num_cols - image = cv2.imread(opt.input) - image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) - height, width = image.shape - cell_width = width / opt.num_cols - cell_height = 2 * cell_width - num_rows = int(height / cell_height) - if num_cols > width or num_rows > height: + 字符列表 = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. " + 字符种数 = len(字符列表) + 列数 = opt.num_cols + 图片 = cv2.imread(opt.input) + 图片 = cv2.cvtColor(图片, cv2.COLOR_BGR2GRAY) + 图高, 图宽 = 图片.shape + 格宽 = 图宽 / opt.num_cols + 格高 = 2 * 格宽 + 行数 = int(图高 / 格高) + if 列数 > 图宽 or 行数 > 图高: print("Too many columns or rows. Use default setting") - cell_width = 6 - cell_height = 12 - num_cols = int(width / cell_width) - num_rows = int(height / cell_height) - - output_file = open(opt.output, 'w') - for i in range(num_rows): - for j in range(num_cols): - output_file.write( - CHAR_LIST[min(int(np.mean(image[int(i * cell_height):min(int((i + 1) * cell_height), height), - int(j * cell_width):min(int((j + 1) * cell_width), - width)]) * num_chars / 255), num_chars - 1)]) - output_file.write("\n") - output_file.close() + 格宽 = 6 + 格高 = 12 + 列数 = int(图宽 / 格宽) + 行数 = int(图高 / 格高) + + 输出文件 = open(opt.output, 'w') + for 行号 in range(行数): + for 列号 in range(列数): + 局部图 = 图片[int(行号 * 格高) : min(int((行号 + 1) * 格高), 图高), + int(列号 * 格宽) : min(int((列号 + 1) * 格宽), 图宽)] + 输出文件.write( + 字符列表[min(int(np.mean(局部图) * 字符种数 / 255), 字符种数 - 1)]) + 输出文件.write("\n") + 输出文件.close() if __name__ == '__main__': -- Gitee From 811cb3529a8ca24e6c7a9f53a5fddaf1ef4fc800 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=B4=E7=83=9C?= Date: Sun, 24 Nov 2024 16:59:25 -0800 Subject: [PATCH 2/5] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E9=87=8D=E6=9E=84:=20?= =?UTF-8?q?=E7=94=9F=E6=88=90=E5=BD=A9=E8=89=B2=E9=83=A8=E5=88=86=E5=91=BD?= =?UTF-8?q?=E5=90=8D,=20=E5=8E=98=E6=B8=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- img2img_color.py | 64 ++++++++++++++++++++++++------------------------ 1 file changed, 32 insertions(+), 32 deletions(-) diff --git a/img2img_color.py b/img2img_color.py index a0818a2..5ab0b8e 100644 --- a/img2img_color.py +++ b/img2img_color.py @@ -25,47 +25,47 @@ def get_args(): def main(opt): if opt.background == "white": - bg_code = (255, 255, 255) + 背景编码 = (255, 255, 255) else: - bg_code = (0, 0, 0) - char_list, font, 字符示例, scale = get_data(opt.language, opt.mode) - num_chars = len(char_list) - num_cols = opt.num_cols - image = cv2.imread(opt.input, cv2.IMREAD_COLOR) - image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) - height, width, _ = image.shape + 背景编码 = (0, 0, 0) + 字符列表, font, 字符示例, scale = get_data(opt.language, opt.mode) + 字符种数 = len(字符列表) + 列数 = opt.num_cols + 图片 = cv2.imread(opt.input, cv2.IMREAD_COLOR) + 图片 = cv2.cvtColor(图片, cv2.COLOR_BGR2RGB) + 图高, 图宽, _ = 图片.shape 字符宽度, 字符高度 = font.getsize(字符示例) 字符高宽比 = 字符高度/字符宽度 - cell_width = width / opt.num_cols - cell_height = 字符高宽比 * cell_width - num_rows = int(height / cell_height) - if num_cols > width or num_rows > height: + 格宽 = 图宽 / opt.num_cols + 格高 = 字符高宽比 * 格宽 + 行数 = int(图高 / 格高) + if 列数 > 图宽 or 行数 > 图高: print("Too many columns or rows. Use default setting") - cell_width = 6 - cell_height = 字符高宽比 * cell_width - num_cols = int(width / cell_width) - num_rows = int(height / cell_height) - out_width = 字符宽度 * num_cols - out_height = scale * 字符高度 * num_rows - out_image = Image.new("RGB", (out_width, out_height), bg_code) - draw = ImageDraw.Draw(out_image) - for i in range(num_rows): - for j in range(num_cols): - partial_image = image[int(i * cell_height):min(int((i + 1) * cell_height), height), - int(j * cell_width):min(int((j + 1) * cell_width), width), :] - partial_avg_color = np.sum(np.sum(partial_image, axis=0), axis=0) / (cell_height * cell_width) - partial_avg_color = tuple(partial_avg_color.astype(np.int32).tolist()) - char = char_list[min(int(np.mean(partial_image) * num_chars / 255), num_chars - 1)] - draw.text((j * 字符宽度, i * 字符高度), char, fill=partial_avg_color, font=font) + 格宽 = 6 + 格高 = 字符高宽比 * 格宽 + 列数 = int(图宽 / 格宽) + 行数 = int(图高 / 格高) + 输出宽度 = 字符宽度 * 列数 + 输出高度 = scale * 字符高度 * 行数 + 输出图片 = Image.new("RGB", (输出宽度, 输出高度), 背景编码) + draw = ImageDraw.Draw(输出图片) + for 行号 in range(行数): + for 列号 in range(列数): + 局部图 = 图片[int(行号 * 格高) : min(int((行号 + 1) * 格高), 图高), + int(列号 * 格宽) : min(int((列号 + 1) * 格宽), 图宽), :] + 局部平均颜色 = np.sum(np.sum(局部图, axis=0), axis=0) / (格高 * 格宽) + 局部平均颜色 = tuple(局部平均颜色.astype(np.int32).tolist()) + 字符 = 字符列表[min(int(np.mean(局部图) * 字符种数 / 255), 字符种数 - 1)] + draw.text((列号 * 字符宽度, 行号 * 字符高度), 字符, fill=局部平均颜色, font=font) if opt.background == "white": - cropped_image = ImageOps.invert(out_image).getbbox() + cropped_image = ImageOps.invert(输出图片).getbbox() else: - cropped_image = out_image.getbbox() - out_image = out_image.crop(cropped_image) - out_image.save(opt.output) + cropped_image = 输出图片.getbbox() + 输出图片 = 输出图片.crop(cropped_image) + 输出图片.save(opt.output) if __name__ == '__main__': -- Gitee From 26886d4a6d82da6751b48398a4124276679faec5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=B4=E7=83=9C?= Date: Sun, 24 Nov 2024 17:00:50 -0800 Subject: [PATCH 3/5] =?UTF-8?q?=E5=86=92=E5=8F=B7=E4=B8=BA=E5=A4=9A?= =?UTF-8?q?=E4=BD=99=3F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- img2img_color.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/img2img_color.py b/img2img_color.py index 5ab0b8e..3d1e386 100644 --- a/img2img_color.py +++ b/img2img_color.py @@ -54,7 +54,7 @@ def main(opt): for 行号 in range(行数): for 列号 in range(列数): 局部图 = 图片[int(行号 * 格高) : min(int((行号 + 1) * 格高), 图高), - int(列号 * 格宽) : min(int((列号 + 1) * 格宽), 图宽), :] + int(列号 * 格宽) : min(int((列号 + 1) * 格宽), 图宽)] 局部平均颜色 = np.sum(np.sum(局部图, axis=0), axis=0) / (格高 * 格宽) 局部平均颜色 = tuple(局部平均颜色.astype(np.int32).tolist()) 字符 = 字符列表[min(int(np.mean(局部图) * 字符种数 / 255), 字符种数 - 1)] -- Gitee From 74bf2385598ffb74c9b8c5b5f2e87871ba30337b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=B4=E7=83=9C?= Date: Sun, 24 Nov 2024 22:09:38 -0800 Subject: [PATCH 4/5] =?UTF-8?q?=E5=88=9D=E6=AD=A5=E5=AE=8C=E6=88=90:=20?= =?UTF-8?q?=E7=94=9F=E6=88=90=E5=BD=A9=E8=89=B2html=E6=A0=BC=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- img2html.py | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 60 insertions(+) create mode 100644 img2html.py diff --git a/img2html.py b/img2html.py new file mode 100644 index 0000000..fd5d3bf --- /dev/null +++ b/img2html.py @@ -0,0 +1,60 @@ +import argparse + +import cv2 +import numpy as np + + +def get_args(): + parser = argparse.ArgumentParser("Image to ASCII") + parser.add_argument("--输入", type=str, default="data/input.jpg", help="原图文件路径") + parser.add_argument("--output", type=str, default="data/output.html", help="生成 html 文件路径") + parser.add_argument("--mode", type=str, default="complex", choices=["simple", "complex"], + help="10 or 70 different characters") + parser.add_argument("--num_cols", type=int, default=150, help="number of character for output's width") + args = parser.parse_args() + return args + +def rgb_to_hex(rgb): + return '#{0:02x}{1:02x}{2:02x}'.format(*rgb) + +def main(opt): + if opt.mode == "simple": + 字符列表 = '@%#*+=-:. ' + else: + 字符列表 = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. " + 字符种数 = len(字符列表) + 列数 = opt.num_cols + 图片 = cv2.imread(opt.输入) + 图片 = cv2.cvtColor(图片, cv2.COLOR_BGR2RGB) + 图高, 图宽, _ = 图片.shape + 格宽 = 图宽 / opt.num_cols + 格高 = 2 * 格宽 + 行数 = int(图高 / 格高) + if 列数 > 图宽 or 行数 > 图高: + print("Too many columns or rows. Use default setting") + 格宽 = 6 + 格高 = 12 + 列数 = int(图宽 / 格宽) + 行数 = int(图高 / 格高) + + 输出文件 = open(opt.output, 'w') + 输出文件.write('
')
+    for 行号 in range(行数):
+        for 列号 in range(列数):
+            局部图 = 图片[int(行号 * 格高) : min(int((行号 + 1) * 格高), 图高),
+                        int(列号 * 格宽) : min(int((列号 + 1) * 格宽), 图宽)]
+            局部平均颜色 = np.sum(np.sum(局部图, axis=0), axis=0) / (格高 * 格宽)
+            局部平均颜色 = tuple(局部平均颜色.astype(np.int32).tolist())
+            输出文件.write(
+                ''
+                + 字符列表[min(int(np.mean(局部图) * 字符种数 / 255), 字符种数 - 1)]
+                + '')
+            
+        输出文件.write("
")
+    输出文件.write('
') + 输出文件.close() + + +if __name__ == '__main__': + opt = get_args() + main(opt) \ No newline at end of file -- Gitee From c0b1a6f66a21e8b9061b7d798c837a1009b653ea Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=B4=E7=83=9C?= Date: Thu, 28 Nov 2024 16:10:44 -0800 Subject: [PATCH 5/5] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=20html=20=E9=83=A8?= =?UTF-8?q?=E5=88=86;=20=E6=B6=A6=E8=89=B2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 9fe657e..cba8fd6 100644 --- a/README.md +++ b/README.md @@ -5,17 +5,23 @@ ## 图片转换为文本 -运行 `python img2txt.py` 读取默认输入 `data/input.jpg`，生成字符化后的文本文件。 +运行 `python img2txt.py`，生成字符化后的文本文件。 + +## 图片转换为 html + +运行 `python img2html.py`，生成字符化后的彩色 html。 ## 图片转换为图片 -运行 `python img2img.py` 读取默认输入 `data/input.jpg`，生成字符化后的黑白图片 `data/output.jpg`。`img2img_color` 生成彩色的。 +运行 `python img2img.py`，生成字符化后的黑白图片 `data/output.jpg`。`img2img_color` 生成彩色的。 加选项用指定语言 (english, german, french ...) 的字符，如` --language french`，暂不支持中日韩（见与原库差别）。 ## 更多选项 -见源码 `get_args` 部分。 +默认读取图片文件 `data/input.jpg` 作为输入，可用 `--输入` 指定其他的文件路径。 + +其他选项见源码 `get_args` 部分。 ## 环境依赖 -- Gitee