Python学习 - Python 图片转字符画

最新推荐文章于 2023-02-03 21:43:29 发布

maxwell0401

最新推荐文章于 2023-02-03 21:43:29 发布

阅读量699

点赞数 1

分类专栏： Python大法

本文链接：https://blog.csdn.net/maxwell0401/article/details/80412573

版权

Python大法专栏收录该内容

2 篇文章 0 订阅

订阅专栏

Python学习 - Python 图片转字符画

来源: https://www.shiyanlou.com/courses/370

原理

pillow 是python的图形处理库
argparse 库是用来管理命令行参数输入的

利用灰度公式

gray ＝ 0.2126 * r + 0.7152 * g + 0.0722 * b

将RGB颜色映射为灰度值, 使用不重复的字符列表, 灰度值小（暗）的用列表开头的符号，灰度值大（亮）的用列表末尾的符号。

代码

#-*-coding:utf-8-*-

from PIL import Image
import argparse

parser = argparse.ArgumentParser()
# 输入文件
parser.add_argument('file')
# 输出文件
parser.add_argument('-o','--output')
# 输出字符画宽度
parser.add_argument('--width',type = int, default = 80)
# 输出字符画高度
parser.add_argument('--height',type = int, default = 80)

# 获取参数
args = parser.parse_args()

IMG = args.file
WIDTH = args.width
HEIGHT = args.height
OUTPUT = args.output

# 定义字符数组
asc_char =  list("$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. ")

# 将rgb值转为字符, 将256灰度映射到70个字符上
def getChar(r,g,b,alpha = 256):
    if alpha == 0:
        return ' '
    length = len(asc_char)
    gray = int(0.2116 * r + 0.7152 * g + 0.0722 * b)

    unit = (256.0 + 1) / length
    return asc_char[int(gray/unit)]

# __name__ 是当前模块名，当模块被直接运行时模块名为 __main__ 
# 这句话的意思就是，当模块被直接运行时，以下代码块将被运行，当模块是被导入时，代码块不被运行。
if __name__ == '__main__':
    im = Image.open(IMG)
    im = im.resize((WIDTH,HEIGHT),Image.NEAREST)

    txt = ""

    for i in range(HEIGHT):
        for j in range(WIDTH):
            txt += getChar(*im.getpixel((j,i)))
        txt += '\n'

    print(txt)

    # 将字符画输出到文件
    with open(OUTPUT if(OUTPUT) else "output.txt",'w') as f:
        f.write(txt)