Get frames from video file, crop and save them

crop_video.py

import cv2

cap = cv2.VideoCapture('video.mp4')
count = 0

while cap.isOpened():
    ret, frame = cap.read()
    if ret:
        h,w = frame.shape[0],frame.shape[1]
        if h > w:
            frame = frame[:w,:w,:]
        else:
            frame = frame[:h,:h,:]
        frame = cv2.resize(frame,(512,512))
        cv2.imwrite('frame{:d}.jpg'.format(count), frame)
        print('writing ','frame{:d}.jpg'.format(count))
        count += 1 # i.e. at 30 fps, this advances one second
        cap.set(1, count)
    else:
        cap.release()
        break