Take screenshot of video python

import cv2
import os
import time

def get_frames(inputFile,outputFolder,step,count):

  '''
  Input:
    inputFile - name of the input file with directoy
    outputFolder - name and path of the folder to save the results
    step - time lapse between each step (in seconds)
    count - number of screenshots
  Output:
    'count' number of screenshots that are 'step' seconds apart created from video 'inputFile' and stored in folder 'outputFolder'
  Function Call:
    get_frames("test.mp4", 'data', 10, 10)
  '''

  #initializing local variables
  step = step
  frames_count = count

  currentframe = 0
  frames_captured = 0

  #creating a folder
  try:  
      # creating a folder named data 
      if not os.path.exists(outputFolder): 
          os.makedirs(outputFolder) 
    
  #if not created then raise error 
  except OSError: 
      print ('Error! Could not create a directory') 
  
  #reading the video from specified path 
  cam = cv2.VideoCapture(inputFile) 

  #reading the number of frames at that particular second
  frame_per_second = cam.get(cv2.CAP_PROP_FPS)

  while (True):
      ret, frame = cam.read()
      if ret:
          if currentframe > (step*frame_per_second):  
              currentframe = 0
              #saving the frames (screenshots)
              name = './data/frame' + str(frames_captured) + '.jpg'
              print ('Creating...' + name) 
              
              cv2.imwrite(name, frame)       
              frames_captured+=1
              
              #breaking the loop when count achieved
              if frames_captured > frames_count-1:
                ret = False
          currentframe += 1           
      if ret == False:
          break
  
  #Releasing all space and windows once done
  cam.release()
  cv2.destroyAllWindows()

Source: stackoverflow.com

Similar codes

0 vote310 views1 answer
Get n largest values from 2D numpy array matrix
Python
0 vote310 views1 answer
Custom jupyter notebook
Python
0 vote311 views1 answer
How to make a tuple
Python
1 vote273 views1 answer
Python get index of item in 2d list
Python
0 vote294 views1 answer
Converting (YYYY-MM-DD-HH:MM:SS) date time
Python
1 vote401 views1 answer
Could not locate a bind configured on mapper mapped class class->tablename, SQL expression or this Session.
Python
1 vote284 views3 answers
Python extract zip file without directory structure
Python
0 vote293 views1 answer
How to navigate to a sub html script selenium python
Python
-1 vote300 views1 answer
How to print the 3erd character of an input in python
Python
24 votes304 views2 answers
How to check a string is empty in python
Python