猫史档案馆

实例：https://shequ.codemao.cn/work/33249665

 

# -*- coding:utf-8 -*-
#coding:utf-8
import argparse
import os
import cv2
import subprocess
from cv2 import VideoWriter, VideoWriter_fourcc, imread, resize
from PIL import Image, ImageFont, ImageDraw
 
# 命令行输入参数处理
# aparser = argparse.ArgumentParser()
# aparser.add_argument('file')
# aparser.add_argument('-o','--output')
# aparser.add_argument('-f','--fps',type = float, default = 24)#帧
# aparser.add_argument('-s','--save',type = bool, nargs='?', default = False, const = True)
# 是否保留Cache文件，默认不保存
 
# 获取参数
# args = parser.parse_args()
# INPUT = args.file
# OUTPUT = args.output
# SAVE = args.save
# FPS = args.fps
# 像素对应ascii码
 
 
ascii_char = list("$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:oa+>!:+. ")
 
 
# ascii_char = list("MNHQ$OC67+>!:-. ")
# ascii_char = list("MNHQ$OC67)oa+>!:+. ")
 
# 将像素转换为ascii码
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)]
 
 
# 将txt转换为图片
def txt2image(file_name):
    im = Image.open(file_name).convert('RGB')
    # gif拆分后的图像，需要转换，否则报错，由于gif分割后保存的是索引颜色
    raw_width = im.width
    raw_height = im.height
    width = int(raw_width / 6)
    height = int(raw_height / 15)
    im = im.resize((width, height), Image.NEAREST)
 
    txt = ""
    colors = []
    for i in range(height):
        for j in range(width):
            pixel = im.getpixel((j, i))
            colors.append((pixel[0], pixel[1], pixel[2]))
            if (len(pixel) == 4):
                txt += get_char(pixel[0], pixel[1], pixel[2], pixel[3])
            else:
                txt += get_char(pixel[0], pixel[1], pixel[2])
        txt += '\n'
        colors.append((255, 255, 255))
 
    im_txt = Image.new("RGB", (raw_width, raw_height), (255, 255, 255))
    dr = ImageDraw.Draw(im_txt)
    # font = ImageFont.truetype(os.path.join("fonts","汉仪楷体简.ttf"),18)
    font = ImageFont.load_default().font
    x = y = 0
    # 获取字体的宽高
    font_w, font_h = font.getsize(txt[1])
    font_h *= 1.37  # 调整后更佳
    # ImageDraw为每个ascii码进行上色
    for i in range(len(txt)):
        if (txt[i] == '\n'):
            x += font_h
            y = -font_w
           # self, xy, text, fill = None, font = None, anchor = None,
            #*args, ** kwargs
        dr.text((y, x), txt[i],  fill=colors[i])
        #dr.text((y, x), txt[i], font=font, fill=colors[i])
        y += font_w
 
    name = file_name
    #print(name + ' changed')
    im_txt.save(name)
 
 
# 将视频拆分成图片
def video2txt_jpg(file_name):
    vc = cv2.VideoCapture(file_name)
    c = 1
    if vc.isOpened():
        r, frame = vc.read()
        if not os.path.exists('Cache'):
            os.mkdir('Cache')
        os.chdir('Cache')
    else:
        r = False
    while r:
        cv2.imwrite(str(c) + '.jpg', frame)
        txt2image(str(c) + '.jpg')  # 同时转换为ascii图
        r, frame = vc.read()
        c += 1
    os.chdir('..')
    return vc
 
 
# 将图片合成视频
def jpg2video(outfile_name, fps):
    fourcc = VideoWriter_fourcc(*"MJPG")
 
    images = os.listdir('Cache')
    im = Image.open('Cache/' + images[0])
    vw = cv2.VideoWriter(outfile_name + '.avi', fourcc, fps, im.size)
 
    os.chdir('Cache')
    for image in range(len(images)):
        # Image.open(str(image)+'.jpg').convert("RGB").save(str(image)+'.jpg')
        frame = cv2.imread(str(image + 1) + '.jpg')
        vw.write(frame)
        print(str(image + 1) + '.jpg' + ' finished')
    os.chdir('..')
    vw.release()
 
 
# 递归删除目录
def remove_dir(path):
    if os.path.exists(path):
        if os.path.isdir(path):
            dirs = os.listdir(path)
            for d in dirs:
                if os.path.isdir(path + '/' + d):
                    remove_dir(path + '/' + d)
                elif os.path.isfile(path + '/' + d):
                    os.remove(path + '/' + d)
            os.rmdir(path)
            return
        elif os.path.isfile(path):
            os.remove(path)
        return
 
 
#获取mp3音频文件代码（请在百度搜索和下载ffmpeg插件，请在程序完毕后自行打开cmd执行）:
#ffmpeg -i 你需要转换的视频路径 -f mp3 需要保存的声音地址
 
#合成音频和视频文:     
#ffmpeg -i 生成出来的字符画avi文件路径 -i  提取完毕的视频声音地址 -strict -2 -f mp4 最终成品保存的路径
 
    INPUT = r"填你需要转换的视频地址"
    OUTPUT = r"F:\New.mp4"#这个无意义，填一个有效的地址就行
    SAVE = r"C:\Users\Administrator\Desktop\video39_3"#这个无意义，填一个有效的地址就行
    vc = video2txt_jpg(INPUT)
    FPS = vc.get(cv2.CAP_PROP_FPS)  # 获取帧率
    vc.release()
    jpg2video(INPUT.split('.')[0], FPS)

#这幅画
from PIL import Image, ImageDraw, ImageSequence


def img2char(imgobj):
    # 图片宽高
    scale = 5
    width, height = (int(imgobj.width/scale), int(imgobj.height/scale))
    # 灰度图
    img_orginal = imgobj.resize((width,height))
    img = imgobj.convert("L").resize((width, height))
    # 70 level 字符串
    ASCII_HIGH = '''$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. '''
    # 创建画布
    img_new = Image.new('RGB', (width * scale, height * scale), 'white')  # 生成空白图
    draw = ImageDraw.Draw(img_new)  # 图片转画布
    # 灰度转字符串
    txt = ''
    for y in range(height):
        for x in range(width):
            pos = (x, y)
            gray = img.getpixel(pos)  # 0-255
            index = int(gray/256*70)
            txt = ASCII_HIGH[index]+' '
            color = img_orginal.getpixel(pos)  # 获取原始颜色
            draw.text((x * scale, y * scale), txt, fill=color)  # 写入文字

    return img_new

# GIF
def gif2char(imgfile):
    img = Image.open(imgfile)
    # 分帧操作
    frames = ImageSequence.Iterator(img)
    # 保存转换后的字符画
    img_colorchars = []
    for frame in frames:
        # 静态图转字符画
        colorchar = img2char(frame)
        img_colorchars.append(colorchar)
    # 合成 GIF
    img_colorchars[0].save("bcm_colorchar.gif",  # 保存路径和文件名
                   append_images=img_colorchars[1:],  # 帧
                   save_all=True,
                   loop=0  # 无限循环
                   )

# main
gif2char("bcm.gif")