Python图片转字符画

七月一号！！！新的一月开始了！！！不能再懒惰了阿！！！！

冲冲冲 这个暑假好好学一下python！！！

 

今天实现简单的项目：图片转字符画

 

主要步骤为：1.安装pillow包

2.命令行输入参数处理并获取参数，目标是获取输入的图片路径、输出字符画的宽和高以及输出文件的路径

3.实现RGB值转字符的函数，首先创建一个字符列表

4.处理图片：

1. 首先使用 PIL 的 Image.open 打开图片文件，获得对象 im
2. 使用 PIL 库的 im.resize() 调整图片大小对应到输出的字符画的宽度和高度，注意这个函数第二个参数使用 Image.NEAREST，表示输出低质量的图片。
3. 遍历提取图片中每行的像素的 RGB 值，调用 getchar 转成对应的字符
4. 将所有的像素对应的字符拼接在一起成为一个字符串 txt
5. 打印输出字符串 txt
6. 如果执行时配置了输出文件，将打开文件将 txt 输出到文件，如果没有，则默认输出到 output.txt 文件

 

完整代码呈上：

# -*- 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 = 60) #输出字符画高

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

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

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

# 将256灰度映射到70个字符上
def get_char(r,g,b,alpha = 256):
    if alpha == 0:
        return ' '
    length = len(ascii_char)
    gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b)

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

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 += get_char(*im.getpixel((j,i)))
        txt += '\n'

    print(txt)

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

 

这里注意几个点：

• RGB函数中alpha值为0时，表示图片此位置空白
• char = get_char(*im.getpixel((j,i)))  这里为什么用*呢？？ 其中 im.getpixel((j,i)) 获取得到坐标 (j,i) 位置的 RGB 像素值（有的时候会包含 alpha 值），返回的结果是一个元组，例如 (1,2,3) 或者 (1,2,3,0)。我们使用 * 可以将元组作为参数传递给 get_char，同时元组中的每个元素都对应到 get_char 函数的每个参数。

 

另外本地通过cmd实现时，需要这样：

 python +py路径+png路径