import cv2import os,sys,shutilimport numpy as np # Open the input movie file, input the filepath asinput_filepath = sys.argv[1]input_movie = cv2.VideoCapture(input_filepath)length = int(input_movie.get(cv2.CAP_PROP_FRAME_COUNT)) #设置outputoutput_movie = cv2.VideoWriter(input_filepath.replace("mp4","avi").replace("input","output"), cv2.VideoWriter_fourcc('D', 'I', 'V', 'X'), 25, (1280, 720)) # Initialize some variablesframe_number = 0 while True: # Grab a single frame of video ret, frame = input_movie.read() frame_number += 1 # Quit when the input video file ends if not ret: break # Draw a box around the body: input the top left point(x,y) and bottom right point(x,y) #cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2) # Write the resulting image to the output video file print("Writing frame {} / {}".format(frame_number, length)) output_movie.write(frame) # All done!input_movie.release()cv2.destroyAllWindows() 
