使用Python和OpenCV从视频中提取图片

前言

OpenCV（Open Source Computer Vision Library）是一个开源的计算机视觉和机器学习库，用于实时计算机视觉。它包含许多用于图像和视频处理的优化算法

在工作中我们有时候需要从视频里面提取关键文本，数据等内容。本文将带你如何使用Python和OpenCV库从视频中提取图片。我们将分步骤讲解以下代码，并讨论如何优化提取过程。

1.我们需要导入所需的库：

import cv2
import os
import sys

2.我们定义图片保存路径：

可以是绝对路径，也可以是相对路径

dstPath =r"./testCut"

3.我们打开视频文件：

cap = cv2.VideoCapture("test.mp4")

如果视频打开失败，我们将打印提示错误消息并退出程序：

if not cap.isOpened():
  print("视频打开失败,退出程序")
  sys.exit()

4.我们使用一个循环来读取视频帧：

frame_count =0


whileTrue:
    #统计计数
    frame_count +=1
    #读取视频帧
    ret, frame = cap.read()

5.每个读取成功的帧，保存为图片：

if ret:
      image_name ="image"+str(frame_count) +".jpg"
      save_path = os.path.join(dstPath,image_name)
      cv2.imwrite(save_path,frame)
    else:
      print("读取帧失败")
      break

6.释放资源并关闭窗口：

cap.release()
cv2.destroyAllWindows()

7.优化提取过程：

1. 按帧间隔提取图片：如果视频帧率较高，可以设置一个间隔，只提取部分帧。例如，每隔10帧提取一帧。

if frame_count %10==0:
    # 保存图片

2. 按时间间隔提取图片：可以设置一个时间间隔，只提取特定时间点的帧。例如，每隔5秒提取一帧。

if frame_count %(5* fps) ==0:
    # 保存图片

3. 按关键帧提取图片：可以使用OpenCV的cv2.VideoCapture.get()函数检查当前帧是否为关键帧，只提取关键帧。

if cap.get(cv2.CAP_PROP_KEYFRAME) ==1:
    # 保存图片

4. 按需求提取图片：可以根据具体需求设置条件，只提取满足条件的帧。例如，可以使用OpenCV的图像处理功能检测特定物体，只提取包含该物体的帧。

完整代码如下

import cv2
import os
import sys


dstPath =r"./testCut"


if not cap.isOpened():
  print("视频打开失败,退出程序")
  sys.exit()




frame_count =0


whileTrue:
    #统计计数
    frame_count +=1
    #读取视频帧
    ret, frame = cap.read()
    if ret:
      image_name ="image"+str(frame_count) +".jpg"
      save_path = os.path.join(dstPath,image_name)
      cv2.imwrite(save_path,frame)
    else:
      print("读取帧失败")
      break




cap.release()
cv2.destroyAllWindows()

关于opencv基本用法，请参考opencv-python，一个超神的Python库这篇文章

总结：

本文介绍了如何使用Python的OpenCV库从视频中提取图片。我们分步骤讲解了代码，并讨论了如何优化提取过程。通过调整参数和条件，可以根据实际需求提取所需的图片。