Python获取视频文件的各种属性信息

很多时候我们需要获取视频文件的属性信息，这里记录一下几种使用Python获取视频文件的属性信息的方法。

方法一：使用opencv库

需要安装opencv库

pip install opencv-python

具体代码如下：

import cv2

def get_video_info_opencv(video_path):
    """
    使用OpenCV获取视频信息
    """
    # 打开视频文件
    cap = cv2.VideoCapture(video_path)
    
    if not cap.isOpened():
        print("无法打开视频文件")
        return None
    
    # 获取视频属性
    fps = cap.get(cv2.CAP_PROP_FPS)           # 帧率
    frame_count = cap.get(cv2.CAP_PROP_FRAME_COUNT)  # 总帧数
    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))   # 宽度
    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)) # 高度
    duration = frame_count / fps              # 时长（秒）
    
    # 关闭视频文件
    cap.release()
    
    return {
        'fps': fps,
        'frame_count': frame_count,
        'width': width,
        'height': height,
        'duration': duration,
        'resolution': f"{width}x{height}"
    }

# 使用示例
video_path = "example.mp4"
info = get_video_info_opencv(video_path)
if info:
    print(f"帧率: {info['fps']}")
    print(f"总帧数: {info['frame_count']}")
    print(f"分辨率: {info['resolution']}")
    print(f"时长: {info['duration']:.2f} 秒")

 

方法二：使用 moviepy

需要安装moviepy库

pip install moviepy

具体代码如下：

from moviepy.editor import VideoFileClip

def get_video_info_moviepy(video_path):
    """
    使用moviepy获取视频信息
    """
    try:
        clip = VideoFileClip(video_path)
        
        info = {
            'fps': clip.fps,
            'frame_count': int(clip.duration * clip.fps),
            'width': clip.size[0],
            'height': clip.size[1],
            'duration': clip.duration,
            'resolution': f"{clip.size[0]}x{clip.size[1]}"
        }
        
        clip.close()
        return info
        
    except Exception as e:
        print(f"处理视频时出错: {e}")
        return None

# 使用示例
video_path = "example.mp4"
info = get_video_info_moviepy(video_path)
if info:
    print(f"帧率: {info['fps']}")
    print(f"总帧数: {info['frame_count']}")
    print(f"分辨率: {info['resolution']}")
    print(f"时长: {info['duration']:.2f} 秒")

 

方法三：使用 cv2 和其他库的综合方案

需要安装opencv库

 

pip install opencv-python

 

具体代码如下：

import cv2
import os

class VideoInfoExtractor:
    def __init__(self, video_path):
        self.video_path = video_path
        self.cap = None
        
    def __enter__(self):
        self.cap = cv2.VideoCapture(self.video_path)
        return self
        
    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.cap:
            self.cap.release()
    
    def get_basic_info(self):
        """获取基础视频信息"""
        if not self.cap or not self.cap.isOpened():
            raise ValueError("无法打开视频文件")
            
        # 获取基本属性
        fps = self.cap.get(cv2.CAP_PROP_FPS)
        frame_count = self.cap.get(cv2.CAP_PROP_FRAME_COUNT)
        width = int(self.cap.get(cv2.CAP_PROP_FRAME_WIDTH))
        height = int(self.cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
        duration = frame_count / fps if fps > 0 else 0
        
        return {
            'fps': fps,
            'frame_count': int(frame_count),
            'width': width,
            'height': height,
            'duration': duration,
            'resolution': f"{width}x{height}",
            'file_size': os.path.getsize(self.video_path)
        }
    
    def get_detailed_info(self):
        """获取详细视频信息"""
        basic_info = self.get_basic_info()
        
        # 获取编码信息
        codec = self.cap.get(cv2.CAP_PROP_CODEC_PIXEL_FORMAT)
        fourcc = self.cap.get(cv2.CAP_PROP_FOURCC)
        
        detailed_info = basic_info.copy()
        detailed_info.update({
            'codec': codec,
            'fourcc': fourcc,
            'aspect_ratio': f"{basic_info['width']}:{basic_info['height']}"
        })
        
        return detailed_info

# 使用示例
video_path = "example.mp4"
try:
    with VideoInfoExtractor(video_path) as extractor:
        info = extractor.get_detailed_info()
        
        print("=== 视频基本信息 ===")
        print(f"文件路径: {video_path}")
        print(f"分辨率: {info['resolution']}")
        print(f"帧率: {info['fps']:.2f} FPS")
        print(f"总帧数: {info['frame_count']}")
        print(f"时长: {info['duration']:.2f} 秒")
        print(f"文件大小: {info['file_size'] / (1024*1024):.2f} MB")
        print(f"宽高比: {info['aspect_ratio']}")
        
except ValueError as e:
    print(f"错误: {e}")