两句代码实现PIL转换为CV2使用的array,另有PyQt5 QImage图像转CV2格式的array

最新推荐文章于 2024-11-02 20:36:26 发布

原创最新推荐文章于 2024-11-02 20:36:26 发布 · 1.4k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#CV2 #python #PIL

python 专栏收录该内容

3 篇文章

订阅专栏

本文介绍如何使用两行代码将PIL图像转换为适用于CV2的数组格式，同时提供从PyQt5的QImage转换为numpy数组的方法，这对于进行图像处理和计算机视觉任务极为实用。

两句代码实现PIL转换为CV2使用的array
PyQt5图像转array(QImage转换为np array)

PIL转换为CV2使用的array

def grab(axis):
    """RGB转BGR"""
    img = ImageGrab.grab(axis)
    img = np.array(img)[..., ::-1]
    return img

QImage转array

def grab_pyqt():
    hw = win32gui.WindowFromPoint(win32api.GetCursorPos())

    app = QApplication(sys.argv)
    screen = QApplication.primaryScreen()
    img = screen.grabWindow(hw).toImage()
    ptr = img.constBits()
    ptr.setsize(img.byteCount())

    mat = np.array(ptr).reshape(img.height(), img.width(), 4)  # 注意这地方通道数一定要填4，否则出错
    return mat