教你将视频指定帧数间隔保存为图像&opencv-python

问题分析

做目标检测、目标跟踪任务中，需要大量数据集进行训练，然后训练的数据是标注好的图像。但是，对于自定义任务时，通常采集的数据集为视频，这就需要对视频指定频帧间隔的图像保存到一个文件夹中，以进行后面的标注任务

环境要求

• python
• opencv-python

实现代码

# 导入所有必要的库
import cv2
import os

# 从指定的路径读取视频
cam = cv2.VideoCapture("F:/video1.mp4")

try:

    # 创建名为data的文件夹
    if not os.path.exists('data'):
        os.makedirs('data')

# 如果未创建，则引发错误
except OSError:
    print('Error: Creating directory of data')


# 定义保存图片函数
# image:要保存的图片名字
# addr；图片地址与相片名字的前部分
# num: 相片，名字的后缀。int 类型
def save_image(image, addr, num):
    address = addr + str(num) + '.jpg'
    cv2.imwrite(address, image)


# reading from frame
ret, frame = cam.read()  # ret为布尔值 frame保存着视频中的每一帧图像 是个三维矩阵

i = 0
timeF = 15  # 设置要保存图像的间隔 15为每隔15帧保存一张图像
j = 0

while ret:
    i = i + 1
    # 如果视频仍然存在，继续创建图像
    if i % timeF == 0:
        # 呈现输出图片的数量
        j = j + 1
        save_image(frame, './data/', j)
        print('save image:', j)
    ret, frame = cam.read()

    # 一旦完成释放所有的空间和窗口
cam.release()
cv2.destroyAllWindows()

执行过程如下：

查看新建文件夹中图像