Как сохранить изображение, прочитанное с помощью camera.capture_continuous (формат rgb) и сохранить его в файл

Question

Я читаю камеру распи через camera.capture_continuous(stream,format='rgb', use_video_port=True, resize=(width, height), чтобы подать ее на USB-ускоритель с краями кораллов. Это работает отлично. Но теперь я хочу сохранить некоторые изображения (в зависимости от анализа) на жесткий диск.

Я начинающий на питоне... file.write не работает. Я предполагаю, потому что я получаю некоторые необработанные данные изображения RGB, а не JPG.

Я хотел бы иметь возможность хранить изображение в формате JPG. Кто-нибудь может подсказать какую функцию использовать?

Обновить

Я попробовал следующее

import argparse
import os
import io
import time
from collections import deque
import numpy as np
import picamera
from PIL import Image
import edgetpu.classification.engine

def main():

  stream = io.BytesIO()
  engine = edgetpu.classification.engine.ClassificationEngine(args.model)

  for foo in camera.capture_continuous(stream,
                                       format='rgb',
                                       use_video_port=True,
                                       resize=(width, height)):
      stream.truncate()
      stream.seek(0)
      input = np.frombuffer(stream.getvalue(), dtype=np.uint8)
      results = engine.ClassifyWithInputTensor(input, top_k=3)

      ...

      image = Image.fromarray(input.astype('uint8'), 'RGB')
      image.save("imgs/image_" + str(i) + ".jpg")

Но только получил ошибку:

Traceback (most recent call last):
  File "mio.py", line 85, in <module>
    main()
  File "mio.py", line 75, in main
    image = Image.fromarray(input.astype('uint8'), 'RGB')
  File "/usr/lib/python3/dist-packages/PIL/Image.py", line 2529, in fromarray
    size = shape[1], shape[0]
IndexError: tuple index out of range

Что я делаю неправильно?

Обновить

Решил это следующей строкой

image = Image.frombuffer('RGB', (width,height), streamValue)

Answer 1

Решил это с помощью следующей строки

image = Image.frombuffer('RGB', (width,height), streamValue)

Answer 2

Вы можете использовать библиотеку подушек для сохранения изображений на диск. Что-то вроде:

pip install Pillow numpy

import numpy as np
from PIL import Image
pixels = np.array([[[255, 0, 0], [0, 255, 0]], [[0, 0, 255], [255, 255, 0]]])
image = Image.fromarray(pixels.astype('uint8'), 'RGB')
image.save('out.jpg')