视频抽帧

做数据集的时候可以通过视频源抽帧获得数据图片

代码部分（需要有Opencv库）

# -*- coding: utf-8 -*-

import shutil
import os
import cv2
import os

videos_src_path = r'./'  # 提取图片的视频文件夹
videos_save_path = r'./'  # 保存图片的路径

videos = os.listdir(videos_src_path)  # 用于返回指定的文件夹包含的文件或文件夹的名字的列表。
videos = filter(lambda x: x.endswith('flv'), videos) #这里可以换成其他格式视频
dirs = os.listdir(videos_src_path)  # 获取指定路径下的文件
for video_name in dirs:  # 循环读取路径下的文件并筛选输出
    if os.path.splitext(video_name)[1] == ".flv":  # 筛选mp4文件
        print(video_name)  # 输出所有的mp4文件
        break

output_img_path = './output8/'  #文件输出路径看一下，可以改成自己的路径
if not os.path.exists(output_img_path):
    os.mkdir(output_img_path)
else:
    # 删除原有目录
    shutil.rmtree(output_img_path)
    # 创建一个新目录
    os.mkdir(output_img_path)

cap = cv2.VideoCapture(video_name)
# videoFPS=cap.get(cv2.CAP_PROP_FPS)
# print (videoFPS)

cnt = 1

print("start")

while (1):
    ret, frame = cap.read()
    # show a frame
    if ret is True:
        print(frame.shape)
        cnt = cnt + 1
        print(cnt)
        if cnt % 50== 0:  #设置每？几秒抽帧
            cropped = frame[0:1920, 0:1920]  # 裁剪坐标为[y0:y1, x0:x1]
            output_name = "%s%s%s" % (output_img_path, str(int(cnt / 10)), '.jpg')
            cv2.imwrite(output_name, cropped)
    else:
        break
print("finished")

cap.release()
cv2.destroyAllWindows()