aXeleRate/example_scripts/tensorflow_lite/detector/camera_pi.py

31 lines
775 B
Python
Executable File

import io
import time
import picamera
import picamera.array
import cv2
from base_camera import BaseCamera
class Camera(BaseCamera):
video_source = 0
@staticmethod
def set_video_source(source):
pass
@staticmethod
def frames():
with picamera.PiCamera(resolution = (1280,720)) as camera:
# let camera warm up
time.sleep(2)
with picamera.array.PiRGBArray(camera, size=(1280,720)) as stream:
while True:
camera.capture(stream, format='bgr', use_video_port=True)
# At this point the image is available as stream.array
image = stream.array
stream.truncate(0)
yield image