一、如何给视频加字幕？

一、如何给视频加字幕？

解题思路：获取视频帧数

二、使用步骤

1.引入库

代码如下（示例）：

import cv2
import random
import numpy as np
import pandas as pd
from PIL import Image, ImageDraw, ImageFont


# 颜色
def rand_color():
    font_r = random.randint(128, 255)
    font_g = random.randint(128, 255)
    font_b = random.randint(128, 255)
    return font_r, font_g, font_b

# 中文读取
data = pd.read_table("chinese_text.txt", header=None)
def chinese_text(intex):
    text_1ist = []
    for x in range(len(data)):
        text_1ist.append(data[0][x])
    return text_1ist[intex]

## 视频保存
def save_video(cap_):
    width_ = int(cap_.get(3))
    height_ = int(cap_.get(4))
    fps_ = cap.get(cv2.CAP_PROP_FPS)
    fourcc_ = cv2.VideoWriter_fourcc(*'mp4v')
    new_video = cv2.VideoWriter(r'../new_video.mp4', fourcc_, fps_, (width_, height_))
    return new_video
    
if __name__ == '__main__':
    path = r'C:\Users\liewei\Desktop\pic\333.mp4'
    cap = cv2.VideoCapture(path)
    new_video_ = save_video(cap)

    fps = cap.get(cv2.CAP_PROP_FPS)
    # total_fps = cap.get(cv2.CAP_PROP_FRAME_COUNT)
    total_fps=cap.get(7)

    current_frame = 0
    while True:
        ret, frame = cap.read()
        if cv2.waitKey(int(np.ceil(total_fps / fps))) & 0xFF == ord('q'):
            break
        elif ret is False:
            break

        for i in range(len(data)):
            start_frame = i * (total_fps / len(data))
            end_frame = (i + 1) * (total_fps / len(data))

            if start_frame <= current_frame <= end_frame:
                img = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_RGB2BGR))
                draw = ImageDraw.Draw(img)
                font = ImageFont.truetype("msyh.ttc", 30, encoding='utf-8')
                draw.text((160, 900), chinese_text(i), font=font, fill=rand_color())
                frame_ = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
                new_video_.write(frame_)
                cv2.imshow('prame', frame_)
            if current_frame > total_fps:
                break
        current_frame += 1
    cap.release()
    new_video_.release()
    cv2.destroyAllWindows()