标签：关键帧 videos python os frame video each path pyav

python使用pyav库提取视频关键帧

pyav库简介

PyAV是[FFmpeg][FFmpeg]库的python绑定。其目标是提供底层库的所有功能和控制，但是尽可能多地管理细节。PyAV用于通过容器、流、包、编解码器和帧直接而精确地访问您的媒体。它公开了该数据的一些转换，并帮助您从其他包(例如Numpy和Pillow)获取数据。具体安装方法参考GitHub

GitHub地址：https://github.com/PyAV-Org/PyAV

官方网站：https://pyav.org/docs/stable/

使用pyav库提取关键帧

代码：

import av
import av.datasets
import os
import cv2
import argparse
import shutil
import math

videos_src_path = './video/'
videos_save_path = './img+/'
if not os.path.exists(videos_save_path):
    os.makedirs(videos_save_path)
file_count = 0
videos = os.listdir(videos_src_path)
for each_video in videos:
    file_count += 1
    each_video_name = each_video.split('.')[0]
    each_video_full_path = os.path.join(videos_src_path, each_video)
    img_path = os.path.join(videos_save_path, each_video_name)
    if os.path.exists(img_path):
        shutil.rmtree(img_path)
    os.mkdir(img_path)
    container = av.open(each_video_full_path)
    # Signal that we only want to look at keyframes.
    stream = container.streams.video[0]
    stream.codec_context.skip_frame = 'NONKEY'

    for frame in container.decode(stream):
        # We use `frame.pts` as `frame.index` won't make must sense with the `skip_frame`.
        frame.to_image().save(img_path + '/' + 'night-sky.{:04d}.jpg'.format(frame.pts), quality=80)


print('\nvideo numbers:', str(file_count), '\nAll is done ,thanks')

其中官方示例：https://pyav.org/docs/stable/cookbook/basics.html#saving-keyframes

标签：关键帧,videos,python,os,frame,video,each,path,pyav
来源： https://blog.csdn.net/lidc1004/article/details/117528327

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。