题目:实现图片转字符串,并通过命令行输入参数
图片:
代码:
from PIL import Image
import argparse
import string
# 创建句柄
parser = argparse.ArgumentParser()
# 添加参数
parser.add_argument('--input','-i')
parser.add_argument('--output', '-o', type = str, default = 'result.txt')
parser.add_argument('--width', '-w', type = int, default = 80)
parser.add_argument('--height', type = int, default = 80)
args = parser.parse_args()
IMG = args.input
WIDTH = args.width
HEIGHT = args.height
FILE = args.output
letters = string.ascii_letters + string.ascii_uppercase + r"!@#$%^&*(){}[]\:;''<>?/~`"
def get_char(r, g, b, alpha = 256):
if alpha == 0:
return ' '
length = len(letters)
gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b)
return letters[int(gray / 256.0 * length)]
if __name__ == '__main__':
img = Image.open(IMG)
img = img.resize((WIDTH, HEIGHT), Image.NEAREST)
char_img = ""
for i in range(HEIGHT):
for j in range(WIDTH):
char_img += get_char(*img.getpixel((j,i)))
char_img += '\n'
print(char_img)
if FILE:
with open(FILE, 'w') as f:
f.write(char_img)