Introducing Python's Douyin Kuaishou character dance

coldplay.xixi
Release: 2021-02-08 10:06:04
forward
2546 people have browsed it

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()
Copy after login

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:
Introducing Pythons Douyin Kuaishou character dance

Dependent application: ffpmeg (download and unzip directly, add the bin directory to the PATH environment variable)
Introducing Pythons Douyin Kuaishou character dance

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:
Introducing Pythons Douyin Kuaishou character dance
Put the original video to be converted in the video and name it test.mp4:
Introducing Pythons Douyin Kuaishou character dance
Open Python3.8
Introducing Pythons Douyin Kuaishou character dance
Run video_2_code_video.py, as shown below, it is running: Introducing Pythons Douyin Kuaishou character dance

will generate some intermediate files such as:
Introducing Pythons Douyin Kuaishou character dance
Introducing Pythons Douyin Kuaishou character dance
After a long wait, finally Get what you want:
Introducing Pythons Douyin Kuaishou character dance

test-txt.mp4 is the code dance you want:
Introducing Pythons Douyin Kuaishou character dance

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!

Related labels:
source:csdn.net
Statement of this Website
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn
Popular Tutorials
More>
Latest Downloads
More>
Web Effects
Website Source Code
Website Materials
Front End Template
About us Disclaimer Sitemap
php.cn:Public welfare online PHP training,Help PHP learners grow quickly!