Python学习day8

学python的小白_ZT

1729人浏览 · 2025-03-16 16:45:08

学python的小白_ZT · 2025-03-16 16:45:08 发布

今天运用所学的知识来尝试解决一下工作中的问题（我的工作是做视频相关的）

首先，我想要通过代码检查一个视频文件的质量，例如有没有丢帧，如果有丢帧的情况，会造成播放视频会卡顿

搜了一些资料，要判断一个视频文件是否丢帧，可借助Python的OpenCV库来实现

第一步是安装OpenCV库，在这一步遇到了挺多困难浪费了很多时间，在此不再对遇到的问题做具体列举，只写一下能成功的方法

第一步：确保正确安装了环境变量

在cmd命令行发送指令python --version，若能正确返回python的版本，则证明环境变量是配置正常的

若未返回python版本，则需要配置一下环境变量

可以在电脑上直接搜环境变量

点击环境变量

在系统变量这里找到Path

添加 Python 安装目录（例如C:\Python39）和 Python 安装目录下的Scripts文件夹路径（例如C:\Python39\Scripts）

保存配置后，在cmd命令行发送指令python --version，若能正确返回python的版本，则证明环境变量配置成功

第二步：在命令行中安装opencv-python库

在cmd中发送指令pip install opencv-python即可成功安装

到此，前期准备结束，接下来是代码部分

1. 打开视频文件：使用cv2.VideoCapture打开视频文件。

2. 逐帧读取视频：使用cap.read()逐帧读取视频，并通过cap.get(cv2.CAP_PROP_POS_FRAMES)获取当前帧的序号。

3. 检查帧序号的连续性：比较当前帧的序号和前一帧的序号，如果不连续，则认为存在丢帧现象。

import cv2


def check_frame_loss(video_path):
    cap = cv2.VideoCapture(video_path)
    if not cap.isOpened():
        print("无法打开视频文件")
        return

    # 获取视频的帧率和总帧数
    fps = cap.get(cv2.CAP_PROP_FPS)
    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))

    # 计算理论时长（秒）
    theoretical_duration = frame_count / fps

    # 实际读取视频帧并计时
    actual_frame_count = 0
    while True:
        ret, frame = cap.read()
        if not ret:
            break
        actual_frame_count += 1

    # 释放资源
    cap.release()

    # 计算实际时长（秒）
    actual_duration = actual_frame_count / fps

    # 判断是否丢帧
    if abs(actual_duration - theoretical_duration) > 0.1:
        print("视频可能存在丢帧现象")
    else:
        print("视频未发现丢帧现象")


if __name__ == "__main__":
    video_file = 'C:\\Users\\zt\\Desktop\\CH1_20250312_133207_000000.ts'
    check_frame_loss(video_file)

找了一个有丢帧的录像文件，运行后的结果如下

能正确检测中视频中的丢帧情况

改一下需求：

1. 检测某个文件夹内所有视频文件是否有丢帧

2. 打印每个文件夹是否有丢帧，总共丢了多少帧

代码如下：

import cv2
import os


def check_frame_loss(video_path):
    cap = cv2.VideoCapture(video_path)
    if not cap.isOpened():
        print(f"无法打开视频文件: {video_path}")
        return 0

    # 获取视频的帧率和总帧数
    fps = cap.get(cv2.CAP_PROP_FPS)
    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))

    # 计算理论时长（秒）
    theoretical_duration = frame_count / fps

    # 实际读取视频帧并计时
    actual_frame_count = 0
    while True:
        ret, frame = cap.read()
        if not ret:
            break
        actual_frame_count += 1

    # 释放资源
    cap.release()

    # 计算实际时长（秒）
    actual_duration = actual_frame_count / fps

    # 计算丢帧数量
    lost_frames = int(abs(frame_count - actual_frame_count))

    return lost_frames


def check_all_videos_in_folder(folder_path):
    total_lost_frames = 0
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            if file.lower().endswith(('.mp4', '.avi', '.mov', '.ts')):
                video_path = os.path.join(root, file)
                lost_frames = check_frame_loss(video_path)
                total_lost_frames += lost_frames
                if lost_frames > 0:
                    print(f"视频 {video_path} 存在丢帧，丢帧数量: {lost_frames}")
                else:
                    print(f"视频 {video_path} 未发现丢帧")

    print(f"文件夹内所有视频总共丢帧数量: {total_lost_frames}")


if __name__ == "__main__":
    folder = 'C:\\Users\\zt\\Desktop\\record'
    check_all_videos_in_folder(folder)

运行后：