-
Notifications
You must be signed in to change notification settings - Fork 30
/
fastframerate.py
80 lines (56 loc) · 2.31 KB
/
fastframerate.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
'''
Created on Sep 9, 2017
@author: inayat
'''
# import the required packages
from imutils.video import FileVideoStream, filevideostream
#from imutils.video import FPS
import numpy as np
import argparse
import imutils
import time
import cv2
from utils.fps2 import FPS2
if __name__ == '__main__':
# Initialize the argument parse which is used to parse the arguments
ap = argparse.ArgumentParser()
ap.add_argument("-v", "--video", required=True,
help="path to input video file")
args = vars(ap.parse_args())
# Since the read method of the openCV cv2.VideoCapture
# is blocking IO operation
# Therefore I am going to use thread enable version of reading
# the video. It is implemented in imutil package (pyImagesearch.com)
# start and open a pointer to the file video stream thread
# and allow the buffer to start to fill
print("[info] starting to read a video file ...")
fvs = FileVideoStream(args["video"]).start()
time.sleep(1.0)
# start the frame per second (FPS) counter
fps = FPS2().start()
# loop over the frames obtained from the video file stream
while fvs.more():
# grab each frame from the threaded video file stream,
# resize
# it, and convert it to grayscale (while still retaining 3
# channels)
frame = fvs.read()
#frame = imutils.resize(frame, width=450)
#frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
#frame = np.dstack([frame, frame, frame])
# display the size of the queue on the frame
#cv2.putText(frame, "Queue Size: {}".format(fvs.Q.qsize()),
# (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2)
fps.update()
cv2.putText(frame, "FPS: {:.2f}".format(fps.fps()),
(10, 50), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2)
# show the frame and update the FPS counter
cv2.imshow("Frame", frame)
cv2.waitKey(1)
# stop the timer and display FPS information
fps.stop()
print("[INFO] elasped time: {:.2f}".format(fps.elapsed()))
print("[INFO] approx. FPS: {:.2f}".format(fps.fps()))
# do a bit of cleanup
cv2.destroyAllWindows()
fvs.stop()