python - 将 PyQt5 QPixmap 转换为 numpy ndarray

Question

我有像素图:

pixmap = self._screen.grabWindow(0,
                                 self._x, self._y,
                                 self._width, self._height)

我想将其转换为 OpenCV 格式。我尝试将其转换为 numpy.ndarray，如 here 所述但我得到错误 sip.voidptr object has an unknown size

有没有办法获取 numpy 数组(与 cv2.VideoCapture read 方法返回的格式相同)？

Answer 1

我使用这段代码得到了 numpy 数组:

channels_count = 4
pixmap = self._screen.grabWindow(0, self._x, self._y, self._width, self._height)
image = pixmap.toImage()
s = image.bits().asstring(self._width * self._height * channels_count)
arr = np.fromstring(s, dtype=np.uint8).reshape((self._height, self._width, channels_count))