opencv - 出现错误 - 'PngImageFile' 对象在使用 Socketio 将视频帧从客户端传输到 Flask 服务器时没有属性 'shape'

Question

我的应用是在客户端打开摄像头，获取帧，在后端对其执行 ML 处理，然后将其返回给客户端。

这部分代码(粗体)抛出错误 - PngImageFile' 对象没有属性 'shape'。

这行代码有问题——frame = imutils.resize(pimg, width=700)

我猜某些处理的格式不正确。请指导

@socketio.on('image')
def image(data_image):
    sbuf = io.StringIO()
    sbuf.write(data_image)

    # decode and convert into image
    b = io.BytesIO(base64.b64decode(data_image))
    pimg = Image.open(b)

    # Process the image frame
    frame = imutils.resize(**pimg,** width=700)
    frame = cv2.flip(frame, 1)
    imgencode = cv2.imencode('.jpg', frame)[1]

    # base64 encode
    stringData = base64.b64encode(imgencode).decode('utf-8')
    b64_src = 'data:image/jpg;base64,'
    stringData = b64_src + stringData

    # emit the frame back
    emit('response_back', stringData)

Answer 1

问题是 pimg 是 PIL 图像格式。而 imutils.resize 函数需要 Numpy 数组格式的图像。因此，在 pimg = Image.open(b) 行之后，您需要将 PIL 图像转换为 Numpy 数组，如下所示:

pimg = np.array(pimg)

为此，您必须像下面这样导入 numpy 库:

import numpy as np