CreateShortVideo/main.py at main · 3dian/CreateShortVideo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#encoding: utf-8

from moviepy.video import fx

from utils.img_utils import *
from utils.video_utils import *
from utils.file_utils import *
from utils.info_utils import *

class Read_Video(object):

    def __init__(self, video_raw_path, desc):
        self.video_raw_path = video_raw_path
        self.desc = desc
        self.video_clip = None
        self.audio = None
        self.video_width = 0
        self.video_height = 0
        self.fps = 25
        self.time = 15

        # 文字字体属性，包含：字体大小、字体颜色、字体位置，一般是上(0)下(1)
        self.font_params = {'size': 50,'color': 'black','kerning': 5,'position': 0}

    def run(self):
        # 1、获取视频的属性数据
        self.__pre()

        # 2、获取视频的某一帧，去PS量取视频的左上角、右下角坐标
        position1, position2 = self.__get_frame(3)

        # 3、对原视频进行一次剪辑，重新产生一个横向的视频
        croped_video_clip = self.video_crop(position1, position2, './source/temp_source_croped.mp4')

        print(f'视频宽：{self.video_width}，高:{self.video_height}，帧率：{self.fps}，时长：{self.time}')

        # 4、一张图片组成背景视频，和视频同帧率、同时长
        image_video_clip = one_pic_to_video('./source/img/mount.jpg', './source/temp_img_video.mp4', self.fps, self.time)

        # 5、合成两段视频
        #synthetic_video_clip = synthetic_video(image_video_clip, croped_video_clip)
        synthetic_video_clip = croped_video_clip
        # 6、生成描述信息
        desc_text_clip = generate_text_clip(self.desc, self.font_params, synthetic_video_clip.duration)

        # 7、视频加入描述信息
        video_with_text_clip = video_with_text(synthetic_video_clip, desc_text_clip)

        # 8、视频加入音频，并删除临时文件
        self.video_with_audio(video_with_text_clip)

    def video_crop(self, position1, position2, croped_video_path):
        """
        视频裁剪
        :return:
        """
        # 裁剪的坐标，包含左上角x轴和y轴；右下角x轴和y轴
        clip2 = fx.all.crop(self.video_clip, x1=position1[0], y1=position1[1], x2=position2[0], y2=position2[1])

        # 保存文件
        clip2.write_videofile(croped_video_path)

        # 时长
        self.time = clip2.duration

        return clip2

    def __pre(self):
        """
        预处理
        :return:
        """
        self.video_raw_clip = VideoFileClip(self.video_raw_path)
        print(self.video_raw_path)
        # 视频宽、高
        self.video_width, self.video_height = self.video_raw_clip.w, self.video_raw_clip.h

        self.fps = self.video_raw_clip.fps
        print(self.fps,self.video_width, self.video_height)

        # 分离出音频
        #self.audio = self.video_raw_clip.audio.subclip(0, self.video_raw_clip.duration - 4)
        self.audio= AudioFileClip("./source/audio/impoth.mp3").subclip(0, self.video_raw_clip.duration - 0)
        # 裁剪尾部的视频素材
        temp_video_clip = self.video_raw_clip.subclip(0, self.video_raw_clip.duration - 0)

        # 生成新的视频,并保存到本地
        temp_video_clip.set_audio(self.audio)

        video_path = './source/temp_source_video.mp4'

        temp_video_clip.write_videofile(video_path, codec='libx264',
                                        audio_codec='aac',
                                        temp_audiofile='temp-audio.m4a',
                                        remove_temp=True)

        # 重新初始化VideoFileClip，便于后面剪辑
        self.video_clip = VideoFileClip(video_path)

    def __get_frame(self, time):
        """
        获取视频的某一帧
        :param time:
        :return:
        """
        # get_frame_from_video(self.path_video_source, time, './output.jpg')

        # 获取要截取视频的左上坐标、右下坐标（328  631）
        # position1 = (0, 328)
        # position2 = (self.video_width, 631)

        # 注意：坐标值必须为偶数，不然会裁剪失败
        position1 = (0, 0)
        position2 = (self.video_width, 1080)

        return position1, position2

    def video_with_audio(self, video_with_text_clip):
        """
        视频合成音频，并删除临时文件
        :return:
        """
        # 设置视频音频，并写入到文件中去
        video_with_text_clip.set_audio(self.audio).write_videofile("output.mp4",
                                                                   codec='libx264',
                                                                   audio_codec='aac',
                                                                   temp_audiofile='temp-audio.m4a',
                                                                   remove_temp=True)
        # 删除所有的临时文件
        #del_temp_file("./source/")


if __name__ == '__main__':
    # 原视频文件
    video_source = './source/video/output.mp4'
    text_info = get_info()

    video_re = Read_Video(video_source, text_info)
    video_re.run()