官网:https://github.com/Breakthrough/PySceneDetect

介绍:

PySceneDetect是一个命令行工具和Python库,用于分析视频,查找场景更改或剪辑。

PySceneDetect集成了外部工具(例如mkvmerge , ffmpeg ),可在使用split-video命令时自动将视频分割为单个片段。还可以为视频生成逐帧分析,称为统计文件,以帮助确定最佳阈值或检测特定视频的模式/其他分析方法。

PySceneDetect使用两种主要的检测方法:detect-threshold (将每个帧与设置的黑电平进行比较,对于检测从黑色到黑色的淡入和淡出有用)和detect-content (比较每个帧,依次查找内容的变化,有用)用于检测视频场景之间的快速切换,尽管处理速度较慢)。每种模式的参数略有不同,并在文档中进行了详细说明.。

通常,如果要使用淡入/淡出/切成黑色来检测场景边界,请使用检测阈值模式。如果视频在内容之间使用大量快速剪切,并且没有明确定义的场景边界,则应使用” 检测内容”模式。一旦知道要使用哪种检测模式,就可以尝试以下建议的参数,或生成统计文件(使用-s / –stats参数),以确定正确的参数-具体来说,是正确的阈值.

安装说明:

PySceneDetect依赖于Python模块numpy,OpenCV(cv2模块)和tqdm(进度条模块,用来显示处理进度),安装命令如下:

  1. pip install scenedetect[opencv]

安装完成后可以通过命令行或代码两种方式进行使用。

使用方法一:命令行

通过命令行cd 目录进入到视频所在路径,然后输入一下命令。

  1. scenedetect --input my_video.mp4 --output my_video_scenes --stats my_video.stats.csv detect-content list-scenes save-images

常用的参数说明如下:

  • –input(-i) :输入视频文件的路径
  • –output(-o) :指定输出目录(可选)
  • –stats:生成统计文件(可选)
  • time:用于设置输入视频持续时间/长度或开始/结束时间。
  • detect-content:切分视频基于内容检测算法。
  • detect-threshold:切分视频基于阈值检测算法。
  • list-scenes:打印场景列表并输出到CSV文件。
  • save-images:为每个场景保存视频中的图像。
  • split-video:使用ffmpeg或mkvMerge对视频进行分割。

完整的参数列表可使用 scenedetect help 命令进行查看。

示例:

  1. # 根据内容算法切割视频(视频片段)
  2. scenedetect -i demo.mp4 detect-content split-video
  3. # 按照内容算法识别视频,并将场景列表输出到终端(CSV文件)
  4. scenedetect -i demo.mp4 time -s 10s detect-content list-scenes
  5. # 根据阈值检测算法,打印场景列表,输出场景图片和视频(常用:导出图片、视频)
  6. scenedetect -i demo.mp4 detect-threshold list-scenes save-images split-video

运行完成后会在当前文件夹生成视频片段,片段截图以及csv文件。

使用方法二:代码

在Python中使用PySceneDetect主要用到下面几个类:

  • VideoManager:用于加载视频并提供搜索;
  • SceneManager:用于协调SceneDetector,VideoManager和可选的StatsManager对象的高级管理器;
  • FrameTimecode:用于存储时间码以及对时间码值进行算术运算(加/减/比较),并具有帧级的精确度;
  • StatsManager:用于存储/缓存帧指标,以加快在同一视频上后续场景检测的运行速度,并可以保存到CSV文件或从CSV中加载缓存;
  • SceneDetector:用于实现检测算法的基类,如ContentDetector,ThresholdDetector等。

官方的示例代码如下(可以配合Comment Translate插件来查看英文注释):

  1. from __future__ import print_function
  2. import os
  3. import scenedetect
  4. from scenedetect.video_manager import VideoManager
  5. from scenedetect.scene_manager import SceneManager
  6. from scenedetect.frame_timecode import FrameTimecode
  7. from scenedetect.stats_manager import StatsManager
  8. from scenedetect.detectors import ContentDetector
  9. STATS_FILE_PATH = 'testvideo.stats.csv'
  10. def main():
  11. # Create a video_manager point to video file testvideo.mp4. Note that multiple
  12. # videos can be appended by simply specifying more file paths in the list
  13. # passed to the VideoManager constructor. Note that appending multiple videos
  14. # requires that they all have the same frame size, and optionally, framerate.
  15. video_manager = VideoManager(['testvideo.mp4'])
  16. stats_manager = StatsManager()
  17. scene_manager = SceneManager(stats_manager)
  18. # Add ContentDetector algorithm (constructor takes detector options like threshold).
  19. scene_manager.add_detector(ContentDetector())
  20. base_timecode = video_manager.get_base_timecode()
  21. try:
  22. # If stats file exists, load it.
  23. if os.path.exists(STATS_FILE_PATH):
  24. # Read stats from CSV file opened in read mode:
  25. with open(STATS_FILE_PATH, 'r') as stats_file:
  26. stats_manager.load_from_csv(stats_file, base_timecode)
  27. start_time = base_timecode + 20 # 00:00:00.667
  28. end_time = base_timecode + 20.0 # 00:00:20.000
  29. # Set video_manager duration to read frames from 00:00:00 to 00:00:20.
  30. video_manager.set_duration(start_time=start_time, end_time=end_time)
  31. # Set downscale factor to improve processing speed.
  32. video_manager.set_downscale_factor()
  33. # Start video_manager.
  34. video_manager.start()
  35. # Perform scene detection on video_manager.
  36. scene_manager.detect_scenes(frame_source=video_manager)
  37. # Obtain list of detected scenes.
  38. scene_list = scene_manager.get_scene_list(base_timecode)
  39. # Like FrameTimecodes, each scene in the scene_list can be sorted if the
  40. # list of scenes becomes unsorted.
  41. print('List of scenes obtained:')
  42. for i, scene in enumerate(scene_list):
  43. print(' Scene %2d: Start %s / Frame %d, End %s / Frame %d' % (
  44. i+1,
  45. scene[0].get_timecode(), scene[0].get_frames(),
  46. scene[1].get_timecode(), scene[1].get_frames(),))
  47. # We only write to the stats file if a save is required:
  48. if stats_manager.is_save_required():
  49. with open(STATS_FILE_PATH, 'w') as stats_file:
  50. stats_manager.save_to_csv(stats_file, base_timecode)
  51. finally:
  52. video_manager.release()
  53. if __name__ == "__main__":
  54. main()