一起用Python做個自動化短視頻生成腳本，實作熱門視頻流水線生產！

前言

前幾天有粉絲和我說，最近在網上看到一些視頻營銷號一天能發布幾百條短視頻， 感覺都是批量生成的，能不能用Python做個自動化短視頻生成腳本呢？

今天就帶大家一起用Python做個自動化視頻生成腳本，實作熱門視頻批量流水線生產，一起來看看吧~

直接跳到文末獲取原始碼福利，

一、核心功能設計

首先我看了網上那些視頻營銷號的視頻，大多數都是圍繞一個主題通過幾張圖片進行視頻輪播展示并添加一些熱門歌曲作為背景音樂，

知道了大概的思路，接下來我們可以通過以下幾步進行實作：

• 通過關鍵字進行圖片批量爬取
• 對爬取的圖片進行篩選并批量進行大小resize統一
• 將統一大小的圖片進行視頻合成
• 根據選取的歌曲，截取合適的背景音樂
• 將視頻和截取的背景音樂合并

二、實作步驟

1. 圖片爬取

當然如果小伙伴已經有視頻素材了，只想給視頻直接添加背景可以直接跳到第四步；如果已經有圖片素材不需要爬取了，可以直接跳到第二步，

第一步，我們可以根據需要的視頻主題進行關鍵字圖片搜索，對圖片進行批量爬取，例如我們以日本動漫《你的名字》作為主題，通過輸入關鍵字，爬取動漫相關的圖片，核心代碼如下：

def dowmloadimg(html, keyword, startNum):
    headers = {'user-agent': 'Mozilla/5.0'}  # 請求頭
    pic_url = re.findall('"objURL":"(.*?)",', html, re.S)  # 找到符合正則規則的網址
    num = len(pic_url)
    i = startNum
    subroot = root
    txtpath = subroot + '/download_img.txt'
    print('找到關鍵詞:' + keyword + '的圖片，開始下載圖片...')
    for each in pic_url:
        a = '第' + str(i + 1) + '張圖片，圖片地址:' + str(each) + '\n'
        b = '正在下載' + a
        print(b)
        path = subroot + '/' + str(i + 1)
        try:
            if not os.path.exists(subroot):
                os.makedirs(subroot)
            if not os.path.exists(path):
                pic = requests.get(each, headers=headers, timeout=10)
                with open(path + '.jpg', 'wb') as f:
                    f.write(pic.content)
                    f.close()
                with open(txtpath, 'a') as f:
                    f.write(a)
                    f.close()
        except:
            traceback.print_exc()
            print('ERROR！！！當前圖片無法下載！！！')
            continue
        i += 1
    return i

words = input("請輸入關鍵字: ")
root = './' + words
if not os.path.exists(root):
    os.makedirs(root)

# 引數為需爬取的頁數
for i in range(2):
    url = 'http://image.baidu.com/search/flip?tn=baiduimage&ie=utf-8&word=' + words + "&pn=" + str(
        pageId) + "&gsm=?&ct=&ic=0&lm=-1&width=0&height=0"
    pageId += 20
    html = requests.get(url, headers=headers)
    lastNum = dowmloadimg(html.text, words, lastNum, )  # 執行一次獲取60張圖

至此我們就可以根據關鍵字將圖片爬取下來，并將圖片根據數字順序命名，如下圖所示，

2. 圖片統一格式大小

第二步，我們需要將獲取的圖片進行大小格式統一，方便后面的視頻合成，這里我們可以自己手動篩選部分需要resize統一大小的圖片，也可以對爬取的整個檔案夾下圖片批量統一格式大小，對于圖片大小尺寸可以通過代碼進行設定，這里我們將所有圖片大小統一成600*800的，核心代碼如下：

# 將所有影像resize成600*800，并保存
def resize_image(image_name):
    # 獲取輸入檔案夾中的所有檔案
    files = os.listdir('./' + image_name)
    output_dir = './resize/'
    # 判斷輸出檔案夾是否存在，不存在則創建
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
    for file in files:
        if file[-4:] == ".jpg":
            img = Image.open('./' + image_name + '/' + file)
            # 將所有圖片均轉成RGB，防止圖片格式例外
            img = img.convert('RGB')
            # resize圖片大小設定成600*800
            img = img.resize((600, 800), Image.ANTIALIAS)
            img.save(os.path.join(output_dir, file))

3. 視頻合成

第三步，我們可以將這些格式大小統一的圖片進行視頻合成，這里我們可以把整個檔案夾中的圖片全部合成，也可以自己手動選取部分，合成的時候我們根據圖片的數字命名順序進行一次合并，核心代碼如下：

def charts2video(img_path, video_path):
    """將給定目錄下的圖片轉成視頻
        img_path: 圖片路徑
        video_path: 輸出視頻的路徑名稱
    回傳: 圖片轉成的視頻
    """
    images = os.listdir(img_path)
    images.sort(key=lambda x: int(x[:-4]))  # 以名稱字串的數字從小到大排序　　
    fps = 1.5  # 幀數
    fourcc = cv2.VideoWriter_fourcc('D', 'I', 'V', 'X')
    im = Image.open(img_path + images[0])
    video_writer = cv2.VideoWriter(video_path, fourcc, fps, im.size)
    for img_i in images:
        frame = cv2.imread(img_path + img_i)
        print('開始將 ' + img_i + ' 加入視頻\n')
        video_writer.write(frame)  # 圖片尺寸必須和視頻尺寸一樣，不然不會被加入視頻中！！！
    video_writer.release()

這里如果我們要控制視頻每秒鐘播放的圖片張數，可以根據fps代表的每一秒播放多少張圖片對幀數自行設定，從而完成圖片的視頻合成，

4. 截取背景音樂

第四步，在背景音樂截取前，我們首先需要確認視頻時長，這樣我們才能截取適時的背景音樂，所以我們可以先獲取視頻的長度，回傳的就是以s為單位的視頻時長，核心代碼如下：

def get_video_time(filename):
  cap = cv2.VideoCapture(filename)
  if cap.isOpened():
    rate = cap.get(5)
    frame_num =cap.get(7)
    duration = frame_num/rate
    return duration
  return -1
t = get_video_duration('./你的名字.mp4')

視頻時長知道了，接下來我們就要進行背景音樂截取了，之前博主有寫過一篇文章，關于如何用Python自制一款炫酷音樂播放器，感興趣的可以去看看是如何制作的，音樂播放器博主已開放原始碼，可通過文末公眾號獲取，

今天我們就使用這款自制的音樂播放器來下載我們需要的音樂，例如我們就以《錯位時空》作為背景音樂，我們可以先通過這款音樂播放器將這首歌下載到本地，如下圖所示，

音樂下載完畢，接下來就是音樂截取，這里我們提供了兩種方式，一種是截取歌曲高潮部分，另一種是自己選取歌曲截取區間，

歌曲高潮自動截取：
經常刷短視頻的小伙伴，可以看到網上很多短視頻背景音樂都是歌曲的高潮部分，那么Python能否自動獲取歌曲的高潮部分呢？當然！Python已經考慮到了，可以自動截取歌曲高潮部分，這里我們需要用到pychorus模塊，具體這個是如何實作，大家可以網上查資料研究研究，

首先我們需要安裝下pychorus模塊

pip install pychorus

安裝完成后，匯入pychorus模塊，一行代碼呼叫就可以實作，這里傳入的引數t就是上面我們獲取的視頻時長，這樣可以確保截取的背景音樂和視頻時長一致，方便視頻和音樂合成，

同時我們還需要注意下，一般一首歌曲的時長大概3-4分鐘，所以我們傳入的t最好不要超過1分鐘，否則可能會出現高潮截取失敗，核心代碼如下：

from pychorus import find_and_output_chorus
chorus_start_sec = find_and_output_chorus("./錯位時空 - 艾辰.mp3", "./錯位時空_high.wav", t)

如下圖，我們知道了視頻的時長是26s，然后截取背景音樂，自動獲取的高潮部分是從2分32秒開始，時長26s，至此我們就可以把當前歌曲的高潮部分截取出來進行最后的視頻和音樂合并了，

自定義截取歌曲區間：

如果本身我們的視頻長度較長，截取不到歌曲高潮部分怎么辦呢？我們可以進行自己定義歌曲截取區間來獲取背景音樂，下面就演示下，截取歌曲從20s開始到90s結束，核心代碼如下：

# 截取背景音樂
audio_background = mpy.AudioFileClip('./錯位時空 - 艾辰.mp3').subclip(20, 90)
audio_background.write_audiofile('bk.mp3')

5. 視頻和背景音樂合并

第五步，我們需要把視頻和剛剛截取好的背景音樂進行合并，合成一個新的視頻，核心代碼如下：

def add_music():
    # 讀取代碼視頻
    my_clip = mpy.VideoFileClip('你的名字.mp4')
    # 截取背景音樂
    audio_background = mpy.AudioFileClip('錯位時空_high.wav')
    # 視頻中插入音頻
    final_clip = my_clip.set_audio(audio_background)
    # 保存最終視頻
    final_clip.write_videofile('result.mp4')

至此，整個視頻就可以自動生成，實作熱門視頻流水線生產了！下面我們一起運行下看看自動生成的視頻效果如何吧，

原始碼及資料已上傳，關注文末公眾號回復【短視頻原始碼】即可獲取完整原始碼

Python往期精彩：

• 見過仙女蹦迪嗎？一起用python做個小仙女代碼蹦迪視頻

• python自制一款炫酷音樂播放器，想聽啥隨便搜！

• 斗地主老是輸？一起用Python做個AI出牌器，豆子蹭蹭漲！

• 一起用Python做個車牌自動識別系統，好玩又實用！

• 桌面太單調？一起用Python做個自定義動態壁紙，竟然還可以放視頻！

• 一起用Python做個自動化彈鋼琴腳本，我竟然彈出了《天空之城》！
  往期精彩原始碼均可通過下方公眾號獲取