python - 生成器无法与 Python 中的字节连接

Question

我有一个类，它使用 get_frame 方法从相机获取帧。在网络环境中，我需要在每个帧周围添加一些数据，然后再将其流式传输到浏览器。当我尝试向帧添加额外信息(一些字节)时，我收到TypeError:无法将字节连接到生成器。如何连接这些数据？

def gen():
    camera = VideoCamera()
    while True:
        frame = camera.get_frame() 
        yield (b'--frame\r\n'
               b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n\r\n')

class VideoCamera():
    def __init__(self):
        self.video = cv2.VideoCapture(0)

    def get_frame(self):        
        while(True):
            ret, frame = self.video.read()
            #that face is the list, which has all the detected faces in the frame, using dlib library
            face = detector(gray, 0)
            for (J, rect) in enumerate(face):
                 ret, jpeg = cv2.imencode('.jpg', frame)
                 yield jpeg.tobytes()

Answer 1

如前所述，调用 get_frame 返回一个生成器，而不是单个帧。您需要迭代该生成器以获取各个帧，然后您可以将其与其他数据一起生成。

def gen():
    camera = VideoCamera()
    for frame in camera.get_frame():
        yield b'--frame\r\nContent-Type: image/jpeg\r\n\r\n'
        yield frame
        yield b'\r\n\r\n'