python画框_python对视频画框标记后保存的方法

需要画框取消注释rectangle

import cv2

import os,sys,shutil

import numpy as np

# Open the input movie file, input the filepath as

input_filepath = sys.argv[1]

input_movie = cv2.VideoCapture(input_filepath)

length = int(input_movie.get(cv2.CAP_PROP_FRAME_COUNT))

#设置output

output_movie = cv2.VideoWriter(input_filepath.replace("mp4","avi").replace("input","output"), cv2.VideoWriter_fourcc('D', 'I', 'V', 'X'), 25, (1280, 720))

# Initialize some variables

frame_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)