Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import os
- import sys
- import cv2
- def main(argv, argc):
- if argc < 2:
- raise Exception('ERROR: No input_video inserted!')
- input_video = argv[1]
- if not os.path.isfile(input_video):
- raise Exception('ERROR: File does not exist!')
- _filename, _ext = os.path.splitext(input_video)
- _basename = os.path.basename(_filename)
- output_dir = _filename
- if argc > 2:
- output_dir = sys.argv[2]
- if not os.path.isdir(output_dir):
- os.makedirs(output_dir)
- vcap = cv2.VideoCapture(input_video)
- cont_frames = 0
- while True:
- cont_frames += 1
- ret, frame = vcap.read()
- if ret:
- img_filename = 'frame_' + str(cont_frames) + '_' + _basename + '.jpg'
- img_filename = os.path.join(output_dir, img_filename)
- cv2.imwrite(img_filename, frame)
- print('saving frame... ' + img_filename)
- else:
- break
- if __name__ == '__main__':
- print('starting app...')
- main(sys.argv, len(sys.argv))
- print('success!')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement