网上找的python的练手项目,用50行代码完成图片转字符画,项目使用了pillow库、argparse库
以下是实现代码,已在linux环境跑过,可正常运行:
from PIL import Image
import argparse
# 外部命令行传入参数处理
parser = argparse.ArgumentParser()
parser.add_argument('file') # 输入文件,外部传入,argparse定位参数用法
parser.add_argument('output') # 传入输出文件路径
parser.add_argument('--width',type = int,default = 80) # 可选参数,宽度
parser.add_argument('--heigh',type = int,default = 80) # 可选参数,高度
# 获取参数
args = parser.parse_args()
IMG = args.file
WIDTH = args.width
HEIGH = args.heigh
OUTPUT = args.output
# 随机字符串表示灰度
ascii_char = list("$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'.")
# 灰度映射函数
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,HEIGH),Image.NEAREST)
output_file = ""
for i in range(HEIGH):
for j in range(WIDTH):
output_file += get_char(*im.getpixel((j,i))) # getpixel输入(x,y) ,输出4个参数
output_file += '\n' # 每行加换行
# 判断是否传入输出文件路径
if OUTPUT:
with open(OUTPUT,'w') as f:
f.write(output_file)
f.close()
else:
print("Can't find output file")