读取本地mp4视频,每隔10帧保存一张图像到本地
import cv2
import os
RootDir = '/Users/billiards'
SaveDir = os.path.join(RootDir,'data')
cam = cv2.VideoCapture(os.path.join(RootDir, '1.mp4'))
try:
if not os.path.exists(SaveDir):
os.makedirs(SaveDir)
except OSError:
print('Error: Creating directory of data')
currentframe = 0
while (True):
ret, frame = cam.read()
if ret:
name = SaveDir + '/frame' + str(currentframe) + '.jpg'
print('Creating...' + name)
if currentframe % 10 == 0:
cv2.imwrite(name, frame)
currentframe += 1
else:
break
cam.release()
cv2.destroyAllWindows()