Python делает снимок экрана и сохраняет его в буфер
Я хочу сделать снимок экрана экрана и сохранить его в буфере как фотографию (X.jpg), а позже я могу использовать cv2(opencv) для чтения того же изображения из буфера.
я действую следующим образом:
Редактировать мой код:
from PIL import ImageGrab
from io import BytesIO
ii = ImageGrab.grab()
with BytesIO() as output:
ii.save(output,format = "JPEG")# This line has an error
cam = output.getvalue()
result, frame = cv2.imencode('.jpg', cam, encode_param)
я получаю эти ошибки:
TypeError: img is not a numpy array, neither a scalar
Спасибо
Можете ли вы объяснить, в чем проблема, и показать все получаемые сообщения об ошибках? Также комментарии в python начинаются с # not //
Возможный дубликат Как сделать снимок экрана в окне или в полноэкранном режиме в Python 3k? (без PIL)
я отредактировал свой вопрос
Если вы уже сохранили его на диск как JPEG, почему бы просто не прочитать содержимое файла непосредственно в буфер (вместо того, чтобы декодировать его с помощью imread, а затем кодировать обратно с помощью imencode)? Или даже лучше, это банальный для преобразования изображения PIL в массив numpy. Тогда просто imencode, а не туда и обратно к файловой системе.
@Dan Mašek, я хочу отправить снимок экрана в другую систему через сокет, я хочу сохранить этот снимок экрана в буфер, чтобы я мог читать и отправлять через сокет, вы говорите, что я могу преобразовать изображение PIL в массив numpy, как? Дело в том, как я могу преобразовать захваченное изображение в массив numpy?
Это объясняется в сообщении, на которое я ссылаюсь. Или вы даже можете просто использовать PIL для кодирования в буфер.
@Dan Mašek Согласно PIL для кодирования в буфер я получаю эту ошибку TypeError: img is not a numpy array, neither a scalar
Если вы изменили свой код и получаете другую ошибку, редактировать ответьте на вопрос и обновите его соответствующим образом.
@Dan Mašek, я редактирую свой вопрос
Позвольте нам продолжить обсуждение в чате.
Последняя строка является избыточной, cam уже является буфером, содержащим поток JPEG, поэтому нет смысла пытаться снова кодировать его с помощью cv2.imencode. Если это save, то ... эй ... может, grab() ничего не получил?
@Dan Mašek grab() работает правильно, потому что, когда я использую ii.save("E:\\aaa.jpg",format = "JPEG",quality=100,optimize=True), я не получаю никаких ошибок, и все работает правильно
Извините, не могу воспроизвести эту проблему локально, работает должным образом.
@Dan Mašek Спасибо за потраченное время
Ответы 1
Вот демонстрация:
from PIL import ImageGrab
from io import BytesIO
import numpy as np
import cv2
## (1) Grab the rgb frame as PIL.Image
ii = ImageGrab.grab()
print(type(ii)) # <class 'PIL.Image.Image'>
## (2) Convert PIL.Image to np.array
rgb = np.array(ii)
print(type(ii)) # <class 'numpy.ndarray'>
## (3) Convert into BGR and display
bgr = cv2.cvtColor(rgb, cv2.COLOR_RGB2BGR)
cv2.imshow("frame", bgr)
cv2.waitKey()
cv2.destroyAllWindows()
пройти через это ссылка на сайт