数据处理——avi视频数据转png图片格式代码，mat转png图片格式代码，图片裁剪成固定大小代码。

Philo`

于 2024-04-28 16:29:41 发布

阅读量1k

点赞数 4

分类专栏：数据预处理文章标签：音视频 python 开发语言

本文链接：https://blog.csdn.net/qq_44864833/article/details/138282582

版权

本文介绍了一个Python脚本，用于读取HMC-QU数据集中的AVI视频，根据Excel文件中的帧范围转换为PNG格式，并对图片进行裁剪。脚本遍历avi文件夹，逐帧抓取并保存为PNG，同时处理可能存在的错误和文件管理问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

avi2png

处理HMC-QU数据集记录，每一个avi视频，通过xlsx中记录的初始帧和结束帧进行帧转png，保存在以avi视频名为名的文件夹下。

import os
import cv2
import pandas as pd

# 输入文件夹路径
folder_path = 'avidata'

# 输出文件夹路径
output_folder = 'avipng'
if not os.path.exists(output_folder):
    os.makedirs(output_folder)

# 读取Excel文件
excel_path = 'du.xlsx'
df = pd.read_excel(excel_path)
def format_numbers(start, end, width=3):
    return [f"{num:0{width}}" for num in range(start, end + 1)]

# 生成png的名字id
save_id = format_numbers(1,1000,4)
# 遍历文件夹下所有avi文件
total = 0
for filename in os.listdir(folder_path):
    if filename.endswith(".avi"):
        video_path = os.path.join(folder_path, filename)
        cap = cv2.VideoCapture(video_path)

        # 检查视频是否成功打开
        if not cap.isOpened():
            print(f"Error: Could not open video {filename}")
            continue

        # 获取视频帧数
        frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))

        video_info = df[df['name'] == filename.split(".")[0