Python自動化短視頻生成腳本實現熱門視頻流水線生產

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張圖

# 將所有圖像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))

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()

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')

pip install pychorus

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

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

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')