python cv读取视频并实时处理输出处理后的视频

# -*- coding: utf-8 -*-

import cv2
import time

import numpy as np
######################     视频载入       #############################
cap = cv2.VideoCapture("E:\\Data_Set\\AODnet\\测试视频\\1.mp4")
fourcc = cv2.VideoWriter_fourcc(*'XVID') 
out = cv2.VideoWriter('E:\\Data_Set\\AODnet\\测试视频\\生成视频\\output11.avi', fourcc, 20, (1920, 1080))
 
#####################       模型载入      #############################

#####################      视频处理       ############################# 
num=0 
while cap.isOpened():
    # get a frame
    rval, frame = cap.read()
    # save a frame
    if rval==True:
      #  frame = cv2.flip(frame,0)
      	# Start time
        start = time.time()
#        rclasses, rscores, rbboxes=process_image(frame) #换成自己调用的函数

        
        clean_image_tensor=process_image(data_hazy) #换成自己调用的函数
        # End time
        


        end = time.time()
    	# Time elapsed
        seconds = end - start+0.0001
        print( "Time taken : {0} seconds".format(seconds))
    	# Calculate frames per second
        fps  = 1 / seconds;
        print( "Estimated frames per second : {0}".format(fps));
        #bboxes_draw_on_img(frame,rclasses,rscores,rbboxes)
        #print(rclasses)
        out.write(clean_image)
        num=num+1
        print(num)
        #fps = cap.get(cv2.CAP_PROP_FPS)
        #print("Frames per second using video.get(cv2.CAP_PROP_FPS) : {0}".format(fps)) 
    else:
        break
    # show a frame
    cv2.imshow("capture", frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
cap.release()
out.release()
cv2.destroyAllWindows()