Introducing Python's Douyin Kuaishou character dance

Free learning recommendation: python video tutorial

First the effect, here’s the video:

Character dance:

Code dance

Source code:

video_2_code_video.py

import argparseimport osimport cv2import subprocessfrom cv2 import VideoWriter_fourccfrom 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文件，默认不保存class Video2CodeVideo:
    def __init__(self):
        self.config_dict = {
            # 原视频文件
            "input_file": "video/test.mp4",
            # 中间文件存放目录
            "cache_dir": "cache",
            # 是否保留过程文件。True--保留，False--不保留
            "save_cache_flag": False,
            # 使用使用的字符集
            "ascii_char_list": list("01B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:oa+>!:+. "),
        }

    # 第一步从函数，将像素转换为字符
    # 调用栈：video_2_txt_jpg -> txt_2_image -> rgb_2_char
    def rgb_2_char(self, r, g, b, alpha=256):
        if alpha == 0:
            return &#39;&#39;
        length = len(self.config_dict["ascii_char_list"])
        gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b)
        unit = (256.0 + 1) / length        return self.config_dict["ascii_char_list"][int(gray / unit)]

    # 第一步从函数，将txt转换为图片
    # 调用栈：video_2_txt_jpg -> txt_2_image -> rgb_2_char
    def txt_2_image(self, file_name):
        im = Image.open(file_name).convert(&#39;RGB&#39;)
        # 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 += self.rgb_2_char(pixel[0], pixel[1], pixel[2], pixel[3])
                else:
                    txt += self.rgb_2_char(pixel[0], pixel[1], pixel[2])
            txt += &#39;\n&#39;
            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] == &#39;\n&#39;):
                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 + &#39; changed&#39;)
        im_txt.save(name)


    # 第一步，将原视频转成字符图片
    # 调用栈：video_2_txt_jpg -> txt_2_image -> rgb_2_char
    def video_2_txt_jpg(self, file_name):
        vc = cv2.VideoCapture(file_name)
        c = 1
        if vc.isOpened():
            r, frame = vc.read()
            if not os.path.exists(self.config_dict["cache_dir"]):
                os.mkdir(self.config_dict["cache_dir"])
            os.chdir(self.config_dict["cache_dir"])
        else:
            r = False
        while r:
            cv2.imwrite(str(c) + &#39;.jpg&#39;, frame)
            self.txt_2_image(str(c) + &#39;.jpg&#39;)  # 同时转换为ascii图
            r, frame = vc.read()
            c += 1
        os.chdir(&#39;..&#39;)
        return vc    # 第二步，将字符图片合成新视频
    def txt_jpg_2_video(self, outfile_name, fps):
        fourcc = VideoWriter_fourcc(*"MJPG")

        images = os.listdir(self.config_dict["cache_dir"])
        im = Image.open(self.config_dict["cache_dir"] + &#39;/&#39; + images[0])
        vw = cv2.VideoWriter(outfile_name + &#39;.avi&#39;, fourcc, fps, im.size)

        os.chdir(self.config_dict["cache_dir"])
        for image in range(len(images)):
            # Image.open(str(image)+&#39;.jpg&#39;).convert("RGB").save(str(image)+&#39;.jpg&#39;)
            frame = cv2.imread(str(image + 1) + &#39;.jpg&#39;)
            vw.write(frame)
            # print(str(image + 1) + &#39;.jpg&#39; + &#39; finished&#39;)
        os.chdir(&#39;..&#39;)
        vw.release()

    # 第三步，从原视频中提取出背景音乐
    def video_extract_mp3(self, file_name):
        outfile_name = file_name.split(&#39;.&#39;)[0] + &#39;.mp3&#39;
        subprocess.call(&#39;ffmpeg -i &#39; + file_name + &#39; -f mp3 -y &#39; + outfile_name, shell=True)

    # 第四步，将背景音乐添加到新视频中
    def video_add_mp3(self, file_name, mp3_file):
        outfile_name = file_name.split(&#39;.&#39;)[0] + &#39;-txt.mp4&#39;
        subprocess.call(&#39;ffmpeg -i &#39; + file_name + &#39; -i &#39; + mp3_file + &#39; -strict -2 -f mp4 -y &#39; + outfile_name, shell=True)

    # 第五步，如果没配置保留则清除过程文件
    def clean_cache_while_need(self):
        # 为了清晰+代码比较短，直接写成内部函数
        def remove_cache_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 + &#39;/&#39; + d):
                            remove_cache_dir(path + &#39;/&#39; + d)
                        elif os.path.isfile(path + &#39;/&#39; + d):
                            os.remove(path + &#39;/&#39; + d)
                    os.rmdir(path)
                    return
                elif os.path.isfile(path):
                    os.remove(path)
                return
        # 为了清晰+代码比较短，直接写成内部函数
        def delete_middle_media_file():
            os.remove(self.config_dict["input_file"].split(&#39;.&#39;)[0] + &#39;.mp3&#39;)
            os.remove(self.config_dict["input_file"].split(&#39;.&#39;)[0] + &#39;.avi&#39;)
        # 如果没配置保留则清除过程文件
        if not self.config_dict["save_cache_flag"]:
            remove_cache_dir(self.config_dict["cache_dir"])
            delete_middle_media_file()

    # 程序主要逻辑
    def main_logic(self):
        # 第一步，将原视频转成字符图片
        vc = self.video_2_txt_jpg(self.config_dict["input_file"])
        # 获取原视频帧率
        fps = vc.get(cv2.CAP_PROP_FPS)
        # print(fps)
        vc.release()
        # 第二步，将字符图片合成新视频
        self.txt_jpg_2_video(self.config_dict["input_file"].split(&#39;.&#39;)[0], fps)
        print(self.config_dict["input_file"], self.config_dict["input_file"].split(&#39;.&#39;)[0] + &#39;.mp3&#39;)
        # 第三步，从原视频中提取出背景音乐
        self.video_extract_mp3(self.config_dict["input_file"])
        # 第四步，将背景音乐添加到新视频中
        self.video_add_mp3(self.config_dict["input_file"].split(&#39;.&#39;)[0] + &#39;.avi&#39;, self.config_dict["input_file"].split(&#39;.&#39;)[0] + &#39;.mp3&#39;)
        # 第五步，如果没配置保留则清除过程文件
        self.clean_cache_while_need()if __name__ == &#39;__main__&#39;:
    obj = Video2CodeVideo()
    obj.main_logic()

Running environment:

Operating system: win10
Version: Python 3.8.4
Dependent library: pip install opencv-python pillow
Installed with administrator rights, mine has been installed, the display is like this:

Dependent application: ffpmeg (download and unzip directly, add the bin directory to the PATH environment variable)

Novice-style operation (big guys, please pretend to be blind):

Name the above source code video_2_code_video.py and create a new file in the same directory Clip video:

Put the original video to be converted in the video and name it test.mp4:

Open Python3.8

Run video_2_code_video.py, as shown below, it is running:

will generate some intermediate files such as:


After a long wait, finally Get what you want:

test-txt.mp4 is the code dance you want:

Lots of free learning recommendations, so please Visit python tutorial(video)

The above is the detailed content of Introducing Python's Douyin Kuaishou character dance. For more information, please follow other related articles on the PHP Chinese website!