视频一键转字符动画——python函数封装和调用练习

from PIL import Image, ImageDraw, ImageFont # pillow库作图
import subprocess   # 执行命令行命令，为了调用ffmpeg
import sys
import os   # 操作目录用
import shutil   # 删除目录用
import numpy as np  # 转化numpy数组
import gc   # 优化运行内存用到

# 定义输入值，这个脚本需要两个输入值：file_input和FPS
file_input = sys.argv[1]
FPS = sys.argv[2]

def do_turn(file_input, FPS):
    # 调用ffmpeg切割视频
    os.makedirs("tempfile/cut/")    # 当前目录新建存放切割图片的临时文件夹
    shell_vedio = "ffmpeg -i " + file_input + " -r " + FPS + " -qscale:v 2 ./tempfile/cut/%05d.jpg"     # 按照XXXXX序号切割
    shell_voice = "ffmpeg -i " + file_input + " ./tempfile/out.mp3"
    subprocess.call(shell_vedio, shell=True)    # 切割视频
    subprocess.call(shell_voice, shell=True)    # 分离音频
    count =0
    for file in os.listdir("./tempfile/cut/"):
        count += 1
    print("成功分离音频，截图开始转换字符画......" + "共计" + str(count) + "张")    # 统计切割图张数

    # 图片转换字符画
    list_p = os.listdir("./tempfile/cut/")  
    cwd = os.getcwd()
    os.mkdir("./tempfile/new/")     # 新建存放字符画的临时文件夹
    process = 1     # 统计完成转换的字符画数量
    for id in list_p:   # 遍历cut文件夹所有切割后的图片做字符画转换
        address = str("".join(cwd + '/tempfile/cut/' + id)) # 拼接文件的绝对路径
        im = Image.open(address)    # 调用image打开图片
        font = ImageFont.truetype("DejaVuSans-Bold", size=20)      # 字体模式，可更改
        rate = 0.1    # 缩放比（不调整的话像素点过多，这里统一调整）
        aspect_ratio = font.getsize("x")[0] / font.getsize("x")[1]  # 获得字符长宽比 
        new_im_size = np.array([im.size[0] * rate, im.size[1] * rate * aspect_ratio]).astype(int)    # 转换numpy数组，调整大小
        im = im.resize(new_im_size)   
        im = np.array(im.convert("L"))       # 转换灰阶图，生成numpy数组
        
        symbols = np.array(list(" .-vM@"))  # 建立字符索引，注意要按照亮度手动排序
        if im.max() == im.min():      # 全黑和全是一种颜色进行区分
            if im.max() > 0:    # 全是一种颜色，亮度值大于0，则全部用最亮的字符数值
                im = (im / im) * (symbols.size - 1)
            else:
                im[np.isnan(im)] = 0    # 全黑时亮度值为NaN(非数值)，则全部用最暗字符的字符数值，也就是全黑
        else: 
            im = (im - im.min()) / (im.max() - im.min()) * (symbols.size - 1) # 根据索引赋予相应像素点相应的数值
        ascii = symbols[im.astype(int)]    
        letter_size = font.getsize("x")   # 获取字符大小，与前面一定要对应
        im_out_size = new_im_size * letter_size    # 这里乘以字符长宽，否则字符只有一个像素点大小
        im_out = Image.new("RGB", tuple(im_out_size), "black")  # 设置输出图片，背景
        draw = ImageDraw.Draw(im_out)   
        
        y = 0   # 两个循环穷举赋值，做字符画图片
        for i, m in enumerate(ascii):
            for j, n in enumerate(m): 
                draw.text((letter_size[0] * j, y), n, font=font) 
            y += letter_size[1]     # 注意+=，这里赋值字符宽度给y值
        im_out.save("./tempfile/new/" + id + ".png")  # 定义输出位置和图片格式
        print(address + "转换成功！当前进度：" + str(process) + "/" + str(count))       # 显示进度
        process += 1
        gc.collect()    # 重要！每次循环结束释放一次内存，否则容易内存溢出
    print("转换成功！开始生成视频，请稍候......")

    # 调用ffmpeg合并字符画为视频，并且合并分离的音频
    outvedio = "ffmpeg -r " + FPS + " -i ./tempfile/new/%05d.jpg.png ./tempfile/out.mp4"
    subprocess.call(outvedio, shell=True)
    final_vedio = "ffmpeg -i ./tempfile/out.mp4 -i ./tempfile/out.mp3 final.mp4"
    subprocess.call(final_vedio, shell=True)
    shutil.rmtree("./tempfile")     # 删除临时文件夹
    print("字符动画final.mp4已生成!已移除临时文件夹")

# 输入格式错误则显示该条用法
def usage():
    print("usage:", sys.argv[0], "<file_input> <FPS>") 
    exit(0)

if __name__ == "__main__":  # 封装，只有在文件作为脚本直接执行时后面的语句才会被执行，而 import 到其他脚本中后面的语句是不会被执行的
    if len(sys.argv) != 3:  # 判断输入的值是否为两个，没错，是判断两个
        usage()
    else:
        do_turn(file_input, FPS)

y = 0   
for i, m in enumerate(ascii):   # 这里i是用不到的
    for j, n in enumerate(m): 
        draw.text((letter_size[0] * j, y), n, font=font)  # 第一个参数是确定坐标
    y += letter_size[1]

symbols = np.array(list(" .-vM@"))      # 可以改成自己想要用的字符，注意按照亮度升序

if im.max() == im.min():
    if im.max() > 0:
        im = (im / im) * (symbols.size - 1)
    else:
        im[np.isnan(im)] = 0
else: 
    im = (im - im.min()) / (im.max() - im.min()) * (symbols.size - 1)