使用 Python和 FFmpeg 批量截图视频到各自文件夹中

news2024/9/22 13:23:10

在这篇博客中，我们将创建一个简单的图形用户界面 (GUI) 工具，利用 wxPython 和 FFmpeg 来从视频文件中批量生成截图。这个工具能够让用户选择一个文件夹，遍历其中的所有视频文件，按照视频长度将其分为四等分，然后为每个视频生成四张截图。所有生成的截图将保存在一个以视频名称命名的文件夹中，并在截图完成后自动打开该文件夹。
C:\pythoncode\new\multivideofilescreenshot.py

工具介绍
  • wxPython：用于创建桌面应用程序的图形界面。
  • FFmpeg：一个强大的多媒体处理工具，用于提取视频帧。

所有代码

import wx
import os
import subprocess
import threading
import datetime
import sys

class VideoScreenshotApp(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self, None, title="视频截图工具", size=(600, 400))

        # 创建面板
        panel = wx.Panel(self)

        # 创建路径选择控件
        self.path_label = wx.StaticText(panel, label="请选择文件夹：")
        self.path_textctrl = wx.TextCtrl(panel, style=wx.TE_READONLY)
        self.path_button = wx.Button(panel, label="选择路径")
        self.path_button.Bind(wx.EVT_BUTTON, self.on_select_path)

        # 创建文件列表控件，使用 CheckListBox 以支持多选
        self.file_list_ctrl = wx.CheckListBox(panel)

        # 创建截图按钮
        self.capture_button = wx.Button(panel, label="截图")
        self.capture_button.Bind(wx.EVT_BUTTON, self.on_capture)

        # 布局
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.path_label, 0, wx.ALL, 5)
        sizer.Add(self.path_textctrl, 0, wx.EXPAND | wx.ALL, 5)
        sizer.Add(self.path_button, 0, wx.ALL, 5)
        sizer.Add(self.file_list_ctrl, 1, wx.EXPAND | wx.ALL, 5)
        sizer.Add(self.capture_button, 0, wx.ALL | wx.ALIGN_CENTER, 5)
        panel.SetSizer(sizer)

        self.current_path = ""

    def on_select_path(self, event):
        dlg = wx.DirDialog(self, "选择文件夹", style=wx.DD_DEFAULT_STYLE)
        if dlg.ShowModal() == wx.ID_OK:
            self.current_path = dlg.GetPath()
            self.path_textctrl.SetValue(self.current_path)
            self.update_file_list()
        dlg.Destroy()

    def update_file_list(self):
        self.file_list_ctrl.Clear()

        if not self.current_path:
            return

        file_list = self.search_video_files(self.current_path)
        for filename, file_path, duration in file_list:
            self.file_list_ctrl.Append(f"{filename} - {str(datetime.timedelta(seconds=int(duration)))}", file_path)

    def search_video_files(self, directory):
        video_extensions = ['.mp4', '.avi', '.mkv', '.mov', '.wmv', '.flv', '.webm']
        file_list = []

        for root, dirs, files in os.walk(directory):
            for file in files:
                if os.path.splitext(file)[1].lower() in video_extensions:
                    file_path = os.path.join(root, file)
                    duration = self.get_video_duration(file_path)
                    file_list.append((file, file_path, duration))

        return file_list

    def get_video_duration(self, file_path):
        cmd = [
            'ffprobe',
            '-v', 'error',
            '-select_streams', 'v:0',
            '-show_entries', 'stream=duration',
            '-of', 'default=noprint_wrappers=1:nokey=1',
            file_path
        ]
        
        try:
            result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True)
            duration_str = result.stdout.strip()
            
            if not duration_str:
                raise ValueError("ffprobe output is empty")

            return float(duration_str)
        
        except subprocess.CalledProcessError as e:
            wx.LogError(f"ffprobe error: {e.stderr}")
            raise
        except ValueError as e:
            wx.LogError(f"Value error: {e}")
            raise

    def on_capture(self, event):
        selected_indices = self.file_list_ctrl.GetCheckedItems()
        if selected_indices:
            for index in selected_indices:
                file_path = self.file_list_ctrl.GetClientData(index)
                file_name = os.path.basename(file_path)
                duration = self.get_video_duration(file_path)
                thread = threading.Thread(target=self.capture_screenshots, args=(file_path, duration))
                thread.start()
        else:
            wx.MessageBox("请先选择一个或多个视频文件", "提示", wx.OK | wx.ICON_INFORMATION)

    def capture_screenshots(self, file_path, duration):
        output_dir = os.path.join(self.current_path, os.path.splitext(os.path.basename(file_path))[0])
        if not os.path.exists(output_dir):
            os.makedirs(output_dir)

        # 计算每张截图的时间点
        intervals = [duration * i / 4 for i in range(1, 5)]

        # 生成截图
        for i, timestamp in enumerate(intervals):
            cmd = [
                'ffmpeg',
                '-ss', str(datetime.timedelta(seconds=int(timestamp))),
                '-i', file_path,
                '-vframes', '1',
                os.path.join(output_dir, f'screenshot_{i+1}.jpg')
            ]
            subprocess.run(cmd, check=True)

        # 截图完成后，自动打开文件夹
        if sys.platform.startswith('win'):
            subprocess.Popen(['explorer', output_dir])
        elif sys.platform.startswith('darwin'):
            subprocess.Popen(['open', output_dir])
        elif sys.platform.startswith('linux'):
            subprocess.Popen(['xdg-open', output_dir])


if __name__ == "__main__":
    app = wx.App(False)
    frame = VideoScreenshotApp()
    frame.Show()
    app.MainLoop()
代码实现

下面是我们的工具实现代码：

import wx
import os
import subprocess
import threading
import datetime
import sys

class VideoScreenshotApp(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self, None, title="视频截图工具", size=(600, 400))

        # 创建面板
        panel = wx.Panel(self)

        # 创建路径选择控件
        self.path_label = wx.StaticText(panel, label="请选择文件夹：")
        self.path_textctrl = wx.TextCtrl(panel, style=wx.TE_READONLY)
        self.path_button = wx.Button(panel, label="选择路径")
        self.path_button.Bind(wx.EVT_BUTTON, self.on_select_path)

        # 创建文件列表控件，使用 CheckListBox 以支持多选
        self.file_list_ctrl = wx.CheckListBox(panel)

        # 创建截图按钮
        self.capture_button = wx.Button(panel, label="截图")
        self.capture_button.Bind(wx.EVT_BUTTON, self.on_capture)

        # 布局
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.path_label, 0, wx.ALL, 5)
        sizer.Add(self.path_textctrl, 0, wx.EXPAND | wx.ALL, 5)
        sizer.Add(self.path_button, 0, wx.ALL, 5)
        sizer.Add(self.file_list_ctrl, 1, wx.EXPAND | wx.ALL, 5)
        sizer.Add(self.capture_button, 0, wx.ALL | wx.ALIGN_CENTER, 5)
        panel.SetSizer(sizer)

        self.current_path = ""

    def on_select_path(self, event):
        dlg = wx.DirDialog(self, "选择文件夹", style=wx.DD_DEFAULT_STYLE)
        if dlg.ShowModal() == wx.ID_OK:
            self.current_path = dlg.GetPath()
            self.path_textctrl.SetValue(self.current_path)
            self.update_file_list()
        dlg.Destroy()

    def update_file_list(self):
        self.file_list_ctrl.Clear()

        if not self.current_path:
            return

        file_list = self.search_video_files(self.current_path)
        for filename, file_path, duration in file_list:
            self.file_list_ctrl.Append(f"{filename} - {str(datetime.timedelta(seconds=int(duration)))}", file_path)

    def search_video_files(self, directory):
        video_extensions = ['.mp4', '.avi', '.mkv', '.mov', '.wmv', '.flv', '.webm']
        file_list = []

        for root, dirs, files in os.walk(directory):
            for file in files:
                if os.path.splitext(file)[1].lower() in video_extensions:
                    file_path = os.path.join(root, file)
                    duration = self.get_video_duration(file_path)
                    file_list.append((file, file_path, duration))

        return file_list

    def get_video_duration(self, file_path):
        cmd = [
            'ffprobe',
            '-v', 'error',
            '-select_streams', 'v:0',
            '-show_entries', 'stream=duration',
            '-of', 'default=noprint_wrappers=1:nokey=1',
            file_path
        ]
        
        try:
            result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True)
            duration_str = result.stdout.strip()
            
            if not duration_str:
                raise ValueError("ffprobe output is empty")

            return float(duration_str)
        
        except subprocess.CalledProcessError as e:
            wx.LogError(f"ffprobe error: {e.stderr}")
            raise
        except ValueError as e:
            wx.LogError(f"Value error: {e}")
            raise

    def on_capture(self, event):
        selected_indices = self.file_list_ctrl.GetCheckedItems()
        if selected_indices:
            for index in selected_indices:
                file_path = self.file_list_ctrl.GetClientData(index)
                file_name = os.path.basename(file_path)
                duration = self.get_video_duration(file_path)
                thread = threading.Thread(target=self.capture_screenshots, args=(file_path, duration))
                thread.start()
        else:
            wx.MessageBox("请先选择一个或多个视频文件", "提示", wx.OK | wx.ICON_INFORMATION)

    def capture_screenshots(self, file_path, duration):
        output_dir = os.path.join(self.current_path, os.path.splitext(os.path.basename(file_path))[0])
        if not os.path.exists(output_dir):
            os.makedirs(output_dir)

        # 计算每张截图的时间点
        intervals = [duration * i / 4 for i in range(1, 5)]

        # 生成截图
        for i, timestamp in enumerate(intervals):
            cmd = [
                'ffmpeg',
                '-ss', str(datetime.timedelta(seconds=int(timestamp))),
                '-i', file_path,
                '-vframes', '1',
                os.path.join(output_dir, f'screenshot_{i+1}.jpg')
            ]
            subprocess.run(cmd, check=True)

        # 截图完成后，自动打开文件夹
        if sys.platform.startswith('win'):
            subprocess.Popen(['explorer', output_dir])
        elif sys.platform.startswith('darwin'):
            subprocess.Popen(['open', output_dir])
        elif sys.platform.startswith('linux'):
            subprocess.Popen(['xdg-open', output_dir])


if __name__ == "__main__":
    app = wx.App(False)
    frame = VideoScreenshotApp()
    frame.Show()
    app.MainLoop()
代码解释

  1. 创建主窗口：

    • 使用 wx.Frame 创建主窗口，添加路径选择控件、文件列表控件以及截图按钮。

  2. 路径选择：

    • on_select_path 方法允许用户选择一个文件夹，并更新文件列表。

  3. 文件列表更新：

    • update_file_list 方法遍历所选文件夹中的视频文件，获取每个视频的时长，并将信息显示在 CheckListBox 中。

  4. 视频时长获取：

    • get_video_duration 方法使用 ffprobe 命令来获取视频时长。

  5. 截图生成：

    • on_capture 方法处理截图请求，使用多线程来生成截图，以避免阻塞主线程。
    • capture_screenshots 方法使用 ffmpeg 命令生成四张截图，并将截图保存在以视频名称命名的文件夹中。

  6. 自动打开文件夹：

    • 截图完成后，自动在文件浏览器中打开保存截图的文件夹。

效果如下

在这里插入图片描述

总结

通过这个工具，你可以轻松地从多个视频文件中生成截图，而无需手动操作。wxPython 提供了一个简单易用的界面，而 FFmpeg 则负责处理视频帧的提取。这个工具不仅对视频编辑工作有帮助，也为批量处理视频文件提供了极大的便利。

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/2057636.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！

相关文章

鸿蒙HarmonyOS之使用ArkTs语言实现自定义Tab菜单栏分页主页面UI

鸿蒙HarmonyOS之使用ArkTs语言实现自定义Tab菜单栏分页主页面UI

一、效果 显示为3个Tab菜单栏，中间可以滑动 二、实现步骤 1、Index.ets 示例代码中用到的颜色、文字、图片等资源可以自行替换 import { Tab_1 } from ./Tab_1; import { Tab_2 } from ./Tab_2; import { Tab_3 } from ./Tab_3;//主页面 Entry Component stru…
阅读更多...
Excel数字中间指定位置插入符号——以120120加*为例

Excel数字中间指定位置插入符号——以120120加*为例

设置单元格格式——自定义 更多阅读Excel数字中间指定位置插入符号_哔哩哔哩_bilibili
阅读更多...
B码对时案例分享，基于RK3568J+Logos-2，让电力设备轻松实现“高精度授时”！

B码对时案例分享，基于RK3568J+Logos-2，让电力设备轻松实现“高精度授时”！

本文主要介绍瑞芯微RK3568J紫光同创Logos-2的B码对时案例，开发环境如下： Windows开发环境：Windows 7 64bit、Windows 10 64bit Pango Design Suite(PDS)：PDS_2022.2-SP3 IRIG-B码对时典型应用 IRIG-B码对时可应用于继电保护装…
阅读更多...
台球厅自动控制无人台球之智能开关-SAAS本地化及未来之窗行业应用跨平台架构

台球厅自动控制无人台球之智能开关-SAAS本地化及未来之窗行业应用跨平台架构

一、无人台球厅优点 1. 降低人力成本：无需雇佣大量员工，如收银员、服务员等，节省了工资、福利和管理成本。 2. 24 小时营业：不受人工营业时间限制，可以随时满足顾客的需求，增加营业时长和收入。 3. 便捷高…
阅读更多...
不良图片检测

不良图片检测

OpenNSFW模型简介 OpenNSFW是一个由 Yahoo 研究院开源的深度学习模型，用于识别和区分网络上的正常内容与不适宜内容（Not Safe For Work）。 项目地址：https://github.com/yahoo/open_nsfw OpenNSFW主要基于Caffe框架实现&#x…
阅读更多...
邮票孔拼版制作方法

邮票孔拼版制作方法

邮票孔拼版制作方法 拼版后的局部图:(中间用连接桥的方式，此方式能最少程度上减少残留) 2）拼版后的效果图 3）邮票孔拼版规则: 拼板与板间距1.2MM或者1.6MM 等邮票孔：8个0.55MM的孔,孔间距0.2MM加两排，邮票孔伸到…
阅读更多...
KubeSphere核心实战_kubesphere部署redis01_为redis指定配置文件_指定存储卷_配置服务---分布式云原生部署架构搭建047

KubeSphere核心实战_kubesphere部署redis01_为redis指定配置文件_指定存储卷_配置服务---分布式云原生部署架构搭建047

然后我们再来,部署一下redis,可以看到,首先去容器官网去找到对应的redis的镜像然后 可以看到镜像中都有说的,如何启动,以及 --appendonly yes 是指定持久化.然后 /data表示数据存储的位置. 可以看到数据存储位置 然后还有配置文件的位置. 可以看到,我们首先去创建配置文件,然后…
阅读更多...
【专题】2024全数驱动 致胜未来-数字化敏捷银行白皮书报告合集PDF分享（附原数据表）

【专题】2024全数驱动 致胜未来-数字化敏捷银行白皮书报告合集PDF分享（附原数据表）

原文链接： https://tecdat.cn/?p37404 政策明确发展使命，新时代商业银行应坚持党建引领，秉持高质量发展理念。数字经济已成大势，商业银行需构建数字基础设施能力，强化顶层战略规划。当前商业银行数字化发展面临诸多挑…
阅读更多...
改变自己·心情治愈

改变自己·心情治愈

今早起床了 看镜子里的我 忽然发现我发型睡得有点kuso 一点点改变，有很大的差别 你我的力量也能改变世界 最近比较烦 最近情绪很down 每天看新闻都会很想大声尖叫 但脏话没用 大家只会嫌凶 我改变自己发现大有不同 新一代的朋友我们好好地加油 大家一起大…
阅读更多...
QT：QWebEngineView基本使用

QT：QWebEngineView基本使用

QWebEngineView介绍 添加库 QT core gui webenginewidgets出现问题 解决 提升类 Widget::Widget(QWidget *parent): QWidget(parent), ui(new Ui::Widget) {ui->setupUi(this);//ui->web_widget->load(QUrl("http://www.baidu.com"));ui->…
阅读更多...
Rimworld边缘世界使用服务器开服联机教程

Rimworld边缘世界使用服务器开服联机教程

订阅联机模组RimWorld Together (MULTIPLAYER)（百度莱卡云） 打开创意工坊搜索RimWorld Together (MULTIPLAYER) 订阅联机模组与它的依赖模组 打开游戏➡点击Mod配置 双击左边灰色的模组，到右边启用 还需要排一下顺序，顺序不对会…
阅读更多...
iPhone变身万能钥匙，iOS 18.1让你的手机解锁一切

iPhone变身万能钥匙，iOS 18.1让你的手机解锁一切

Apple 近日发布新闻稿宣布，在即将推出的 iOS 18.1 更新中，将开放全新的 NFC（近场通信）和 SE（安全元件）API 给第三方开发者使用。这一举措将使得开发者能够开发出更加多样化和实用的应用程序，如车…
阅读更多...
Redis清空缓存

Redis清空缓存

Windows环境下使用命令行进行redis缓存清理 1、redis安装目录下输入cmd 2、redis-cli -p 端口号 3、flushdb 清除当前数据库缓存 4、flushall 清除整个redis所有缓存
阅读更多...
Dubbo源码深度解析(七)

Dubbo源码深度解析(七)

接上一篇博客《Dubbo源码深度解析(六)》，上篇博客主要从服务消费方开始讲起，主要讲：如果类中的属性或者方法，如果被DubboReference注解所修饰，Dubbo是怎么处理的，处理逻辑类似Spring框架提供的Autowired注解…
阅读更多...
JavaScript - Ajax

JavaScript - Ajax

Asynchronous JavaScript And XML，异步的JavaScript和XML 作用: 数据交换：通过Ajax可以给服务器发送请求，并获取服务器响应的数据。异步交互：可以在不重新加载整个页面的情况下，与服务器交换数据并更新部分网页的技术…
阅读更多...
Vue2 与 Vue3 生命周期对比

Vue2 与 Vue3 生命周期对比

首先我们需要了解 Vue2 的生命周期，请参考我之前写的博文 Vue2-生命周期 Vue2 生命周期存在四个阶段，八个钩子函数，具体作用可以看图，详细解释可以参考之前的博文 Vue3 生命周期 经过对比，我们发现 Vue3 的生命周期也存…
阅读更多...
Office办公软件从哪里下载

Office办公软件从哪里下载

Offece下载方法有哪些？从哪里下载Office办公软件？是日常办公必不可少的常用软件。 进入正题 第一步：下载最新的office办公软件软件包 https://docs.qq.com/doc/DQUhtSnJxU1FxZnVF 先下载后解压，看到下面的文件 然后右击-以管理…
阅读更多...
mq-work模型

mq-work模型

work模型是什么-任务模型 会产生什么现象？ 操作 1、新建一个队列 2、在consuner增加2个消费者 2、发送50条消息 4、消费者1和消费者2，一轮1次 消费能力不一致，还是一人一个消费50条消息，所以浪费了能力 总结 prefetch预获取&#…
阅读更多...
文心快码 Baidu Comate 前端工程师观点分享：以文心快码 Baidu Comate为例，智能代码助手需要什么（一）

文心快码 Baidu Comate 前端工程师观点分享：以文心快码 Baidu Comate为例，智能代码助手需要什么（一）

本系列视频来自百度工程效能部的前端研发经理杨经纬，她在由开源中国主办的“AI编程革新研发效能”OSC源创会杭州站105期线下沙龙活动上，从一款文心快码（Baidu Comate）前端工程师的角度，分享了关于智能研发工具本身的研…
阅读更多...
单片机原理及技术（六）—— 中断系统的工作原理

单片机原理及技术（六）—— 中断系统的工作原理

目录 一、AT89S51中断技术概述 二、AT89S51中断系统结构 2.1 中断请求源 2.2 中断请求标志寄存器 2.2.1 TCON 寄存器 2.2.2 SCON 寄存器 三、中断允许与中断优先级的控制 3.1 中断允许寄存器 IE 3.2 中断优先级寄存器 IP 四、响应中断请求的条件 五、外部中断的触发…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023