python Opencv目录下视频转换为

import cv2
import os

curdir="/home/lisa/dataset/video/selected/"

index=1;

interval=10


def video2pic(path):
    global  interval
    cap=cv2.VideoCapture(path);
    success,frame=cap.read()
    while success:
        global index
        index=index+1
        if index%interval!=0:
            success, frame = cap.read()
            continue
        else:
            temp=index/interval

            cv2.imshow("temp",frame)
            if index<10000:
                cv2.imwrite("/home/lisa/dataset/handPicStage2/720p/1/"+str(temp)+".jpg",frame)
            else:
                cv2.imwrite("/home/lisa/dataset/handPicStage2/720p/2/" + str(temp) + ".jpg", frame)
            success, frame = cap.read()
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break

for parent,dirname,filenames in os.walk(curdir):
    for filename in filenames:
        path=os.path.join(parent,filename)
        if path.endswith("mp4"):
            video2pic(path)
    pass