Extract images from videos that saved in same directory.

capture_video.py

import cv2
import math
import os

video_files = next(os.walk('./'), (None, None, []))[2]

for video in video_files:
	print(video)
	cap = cv2.VideoCapture(video)
	frameRate = cap.get(5) #frame rate
	x=1
	if not os.path.isdir(video):
    file_path = os.path.splitext(video)[0]
		file_name = file_path.split('/')[-1]
		os.makedirs(file_name)
	while(cap.isOpened()):
		frameId = cap.get(1) #current frame number
		ret, frame = cap.read()
		if (ret != True):
			break
		if (frameId % math.floor(frameRate) == 0):
			filename = './'+file_name+'/image' +  str(int(x)) + ".jpg";x+=1
			print(filename)
			cv2.imwrite(filename, frame)
	cap.release()
print ("Done!")