使用16位图像数据的Image.frombuffer

Question

如果我的Windows系统是32位颜色深度模式，那么下面的代码可以从一个窗口获取到一个不错的PIL图像：

def image_grab_native(window):
    hwnd = win32gui.GetDesktopWindow()

    left, top, right, bot = get_rect(window)
    w = right - left
    h = bot - top

    hwndDC = win32gui.GetWindowDC(hwnd)
    mfcDC  = win32ui.CreateDCFromHandle(hwndDC)
    saveDC = mfcDC.CreateCompatibleDC()

    saveBitMap = win32ui.CreateBitmap()
    saveBitMap.CreateCompatibleBitmap(mfcDC, w, h)

    saveDC.SelectObject(saveBitMap)
    saveDC.BitBlt((0, 0), (w, h),  mfcDC,  (left, top),  win32con.SRCCOPY)

    bmpinfo = saveBitMap.GetInfo()
    bmpstr = saveBitMap.GetBitmapBits(True)

    im = Image.frombuffer(
        'RGB',
        (bmpinfo['bmWidth'], bmpinfo['bmHeight']),
        bmpstr, 'raw', 'BGRX', 0, 1)

    win32gui.DeleteObject(saveBitMap.GetHandle())
    saveDC.DeleteDC()
    mfcDC.DeleteDC()
    win32gui.ReleaseDC(hwnd, hwndDC)

    return im

但是，当我在16位模式下运行时，就会出现错误：

>>> image_grab_native(win)
Traceback (most recent call last):
  File "<pyshell#3>", line 1, in <module>
    image_grab_native(win)
  File "C:\claudiu\bumhunter\finderbot\ezpoker\utils\win32.py", line 204, in image_grab_native
    bmpstr, 'raw', 'BGRX', 0, 1)
  File "c:\python25\lib\site-packages\PIL\Image.py", line 1808, in frombuffer
    return apply(fromstring, (mode, size, data, decoder_name, args))
  File "c:\python25\lib\site-packages\PIL\Image.py", line 1747, in fromstring
    im.fromstring(data, decoder_name, args)
  File "c:\python25\lib\site-packages\PIL\Image.py", line 575, in fromstring
    raise ValueError("not enough image data")
ValueError: not enough image data

我该如何构造frombuffer的调用，以便在16位模式下正常工作？另外，我怎么才能让这个函数在任何位深度模式下都能工作，而不是每次都得传一个参数呢？

更新：从这个问题中我了解到，我必须使用"BGR;16"而不是"BGRX"作为第二个模式参数。这样可以正确获取图像，无论是否指定步幅。问题是某些像素值稍微有点偏差：

x   y native           ImageGrab
280 0  (213, 210, 205) (214, 211, 206)
280 20 (156, 153, 156) (156, 154, 156)
280 40 (213, 210, 205) (214, 211, 206)
300 0  (213, 210, 205) (214, 211, 206)

这是从同一个窗口中取的一些值样本。截图在肉眼看起来是一样的，但我需要对一些像素进行处理……另外，我想使用本地方法的原因是它稍微快一点，而且在双显示器的虚拟机中运行时表现更好……（是的，听起来有点复杂，我知道）。

图像处理 PIL库像素值 16位图像颜色深度 BGR模式图像获取虚拟机性能

使用16位图像数据的Image.frombuffer

1 个回答

撰写回答