jens
5b73cd0e1d
git-svn-id: http://moon:8086/svn/software/trunk/projects/opencv@310 b431acfa-c32f-4a4a-93f1-934dc6c82436
65 lines
1.8 KiB
Python
Executable File
65 lines
1.8 KiB
Python
Executable File
# import the necessary packages
|
|
from __future__ import print_function
|
|
from imutils.video.pivideostream import PiVideoStream
|
|
from imutils.video import FPS
|
|
from picamera.array import PiRGBArray
|
|
from picamera import PiCamera
|
|
import argparse
|
|
import imutils
|
|
import time
|
|
import cv2
|
|
|
|
# construct the argument parse and parse the arguments
|
|
ap = argparse.ArgumentParser()
|
|
ap.add_argument("-n", "--num-frames", type=int, default=100,
|
|
help="# of frames to loop over for FPS test")
|
|
ap.add_argument("-d", "--display", type=int, default=-1,
|
|
help="Whether or not frames should be displayed")
|
|
args = vars(ap.parse_args())
|
|
|
|
# initialize the camera and stream
|
|
camera = PiCamera()
|
|
camera.resolution = (320, 240)
|
|
camera.framerate = 32
|
|
rawCapture = PiRGBArray(camera, size=(320, 240))
|
|
stream = camera.capture_continuous(rawCapture, format="bgr",
|
|
use_video_port=True)
|
|
|
|
# allow the camera to warmup and start the FPS counter
|
|
print("[INFO] sampling frames from `picamera` module...")
|
|
time.sleep(2.0)
|
|
fps = FPS().start()
|
|
|
|
# loop over some frames
|
|
for (i, f) in enumerate(stream):
|
|
# grab the frame from the stream and resize it to have a maximum
|
|
# width of 400 pixels
|
|
frame = f.array
|
|
frame = imutils.resize(frame, width=400)
|
|
|
|
# check to see if the frame should be displayed to our screen
|
|
if args["display"] > 0:
|
|
cv2.imshow("Frame", frame)
|
|
key = cv2.waitKey(1) & 0xFF
|
|
|
|
# clear the stream in preparation for the next frame and update
|
|
# the FPS counter
|
|
rawCapture.truncate(0)
|
|
fps.update()
|
|
|
|
# check to see if the desired number of frames have been reached
|
|
if i == args["num_frames"]:
|
|
break
|
|
|
|
# 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()
|
|
stream.close()
|
|
rawCapture.close()
|
|
camera.close()
|
|
|