python - 在python + picamera + opencv中的线程内使用 “with”

我将树莓派与picamera和opencv python模块一起使用，试图进行一些快速捕获和处理。
目前，我正在使用http://picamera.readthedocs.org/en/latest/recipes2.html#rapid-capture-and-processing中的配方将每个图像捕获到BytesIO流中。然后，我在ImageProccessor类中添加了代码，以将每个流转换为opencv对象，并“即时”进行一些分析。

因此，我当前的代码如下所示:

import io
import time
import threading
import picamera
import cv2
import picamera.array
import numpy as np


# Create a pool of image processors
done = False
lock = threading.Lock()
pool = []

class ImageProcessor(threading.Thread):
    def __init__(self):
        super(ImageProcessor, self).__init__()
        self.stream = io.BytesIO()
        self.event = threading.Event()
        self.terminated = False
        self.start()

    def run(self):
        # This method runs in a separate thread
        global done
        while not self.terminated:
            # Wait for an image to be written to the stream
            if self.event.wait(1):
                try:
                    self.stream.seek(0)
                    # Read the image and do some processing on it
                    # Construct a numpy array from the stream
                    data = np.fromstring(self.stream.getvalue(), dtype=np.uint8)
                    # "Decode" the image from the array, preserving colour
                    image = cv2.imdecode(data, 1)

                    # Here goes more opencv code doing image proccessing

                    # Set done to True if you want the script to terminate
                    # at some point
                    #done=True
                finally:
                    # Reset the stream and event
                    self.stream.seek(0)
                    self.stream.truncate()
                    self.event.clear()
                    # Return ourselves to the pool
                    with lock:
                        pool.append(self)

def streams():
    while not done:
        with lock:
            if pool:
                processor = pool.pop()
            else:
                processor = None
        if processor:
            yield processor.stream
            processor.event.set()
        else:
            # When the pool is starved, wait a while for it to refill
            print ("Waiting")
            time.sleep(0.1)

with picamera.PiCamera() as camera:
    pool = [ImageProcessor() for i in range(4)]
    camera.resolution = (640, 480)
    camera.framerate = 30
    camera.start_preview()
    time.sleep(2)
    camera.capture_sequence(streams(), use_video_port=True)

# Shut down the processors in an orderly fashion
while pool:
    with lock:
        processor = pool.pop()
    processor.terminated = True
    processor.join()

问题在于，这涉及到每个图像的JPEG编码和解码，这是有损且耗时的。建议的替代方法是捕获到picamera.ar​​ray:http://picamera.readthedocs.org/en/latest/recipes1.html#capturing-to-an-opencv-object，对于单个图像，代码:

import time
import picamera
import picamera.array
import cv2

with picamera.PiCamera() as camera:
    camera.start_preview()
    time.sleep(2)
    with picamera.array.PiRGBArray(camera) as stream:
        camera.capture(stream, format='bgr')
        # At this point the image is available as stream.array
        image = stream.array

效果很好，但我不知道如何组合这两段代码，以便ImageProcessor类定义picamera.ar​​ray而不是BytesIO流。需要使用“with”语句为picamera.ar​​ray生成流使我感到困惑(我是python ...;的新手)。
感谢您的指导。
天使

最佳答案

我发现您可以仅从picamera模块引用源。

def raw_resolution(resolution):
    """
    Round a (width, height) tuple up to the nearest multiple of 32 horizontally
    and 16 vertically (as this is what the Pi's camera module does for
    unencoded output).
    """
    width, height = resolution
    fwidth = (width + 31) // 32 * 32
    fheight = (height + 15) // 16 * 16
    return fwidth, fheight


def bytes_to_rgb(data, resolution):
    """
    Converts a bytes objects containing RGB/BGR data to a `numpy`_ array.
    """
    width, height = resolution
    fwidth, fheight = raw_resolution(resolution)
    if len(data) != (fwidth * fheight * 3):
        raise PiCameraValueError(
            'Incorrect buffer length for resolution %dx%d' % (width, height))
    # Crop to the actual resolution
    return np.frombuffer(data, dtype=np.uint8).\
            reshape((fheight, fwidth, 3))[:height, :width, :]

您可以通过致电进行转换

image = bytes_to_rgb(self.stream.getvalue(),resolution)

分辨率为(宽度，高度)。 camera传递给PiRGBArray的原因是能够引用相机的分辨率。

关于python - 在python + picamera + opencv中的线程内使用 “with”，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/29065624/