Update video.py

video.py

@@ -6,11 +6,14 @@ from yolo import YOLO
 from PIL import Image
 import numpy as np
 import cv2
+import time
 yolo = YOLO()
 # 调用摄像头
-capture=cv2.VideoCapture(1) # capture=cv2.VideoCapture("1.mp4")
+capture=cv2.VideoCapture(0) # capture=cv2.VideoCapture("1.mp4")
 
+fps = 0.0
 while(True):
+    t1 = time.time()
     # 读取某一帧
     ref,frame=capture.read()
     # 格式转变，BGRtoRGB
@@ -23,11 +26,15 @@ while(True):
 
     # RGBtoBGR满足opencv显示格式
     frame = cv2.cvtColor(frame,cv2.COLOR_RGB2BGR)
+    
+    fps  = ( fps + (1./(time.time()-t1)) ) / 2
+    print("fps= %.2f"%(fps))
+    frame = cv2.putText(frame, "fps= %.2f"%(fps), (0, 40), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
+    
     cv2.imshow("video",frame)
     c= cv2.waitKey(30) & 0xff 
     if c==27:
         capture.release()
         break
 
-yolo.close_session()
-    
\ No newline at end of file
+yolo.close_session()