opencv 抽取无人机影像视频帧及对应坐标参数

基于opencv-python ，依据设置的帧数间隔提取关键帧，以及解析无人机字幕数据，生成csv

import os
import cv2
from tqdm import tqdm
import pysrt
import csv
import re

#r"D:\2024\2024_12\D3509122B001S\srt\DJI_20241211133017_0004_V.SRT"
srt_path = input("1.请输入srt文件路径：")
#r"D:\2024\2024_12\D3509122B001S\mp4\1界线-沙塘澳_20241211133017_0004_V.MP4"
mp4_path = input("2.请输入mp4文件路径：")
#"20241211133017_0004_V" #不要加中文
file_name = input("3.请输入保存文件名,不要有中文：")
#r"D:\2024\2024_12\D3509122B001S\img"
save_path = input("4.请输入保存路径：")

while True:
    pattern = r"\[(.*?)\]"
    #001_002
    print("5.进入抽取模式，请配置关键参数\n")

    timeRate = float(input("请输入时间间隔（秒）：")) #以秒为单位
    frameRate = int(timeRate * 30)  # 帧数截取间隔（每隔30帧截取一帧,一秒30帧）
    cut_rat = float(input("请输入裁剪比例（如0.167）："))#1/6 # 裁剪比例
    try:
        txt_data = []
        #读取srt文件
        srt_data = pysrt.open(srt_path)
        for i, item in enumerate(srt_data):
            if(i%frameRate == 0):
                # 获取字幕的起始时间和结束时间
                start_time = item.start
                hour = str(start_time.hours)
                minute = str(start_time.minutes)
                second = str(start_time.seconds)
                microsecond = str(start_time.milliseconds)
                text = item.text.split('\n')
                time = text[1]
                matches = re.findall(pattern, text[2])
                latitude = matches[8].split(':')[1].strip()
                longitude = matches[9].split(':')[1].strip()
                rel_alt = matches[10].split(' ')[1].strip()
                abs_alt = matches[10].split(' ')[3].strip()
                gd_yaw = matches[11].split(' ')[1].strip()
                gd_pitch = matches[11].split(' ')[3].strip()
                gd_roll = matches[11].split(' ')[5].strip()
                txt_data.append([hour, minute, second,
                                microsecond, latitude, longitude,
                                    rel_alt, abs_alt, gd_yaw, gd_pitch, gd_roll, time])
                # 将时间转换为秒

        c = 0
        cap = cv2.VideoCapture(mp4_path)
        while(True):
            ret, frame = cap.read()
            if ret:
                if(c % frameRate == 0):
                # 这里就可以做一些操作了：显示截取的帧图片、保存截取帧到本地
                # 从顶部裁剪1/6
                    frame = frame[int(cut_rat*frame.shape[0]):, :]
                    #str(c)补全四位数
                    save_name = file_name + '_'+ str(c).zfill(4) + '.jpg'
                    cv2.imwrite(save_path + "/" + save_name, frame)  # 这里是将截取的图像保存在本地
                    txt_data[c//frameRate].append(save_name)
                c += 1
                cv2.waitKey(0)
            else:
                print("所有帧都已经保存完成")
                break
        cap.release()
        #txt_data 写为csv

        with open(save_path + "/" + file_name + ".csv", "w", newline='') as f:
            #写入标题行
            f.write("hour,minute,second,microsecond,latitude,longitude,rel_alt,abs_alt,gd_yaw,gd_pitch,gd_roll,time,img_name\n")
            writer = csv.writer(f)
            writer.writerows(txt_data)
    except Exception as e:
        print(e)