我想進行螢屏截圖而不是將其轉換為陣列而不將檔案保存為路徑中的影像并從路徑中再次加載以進行轉換：
我想要的是直接將資料轉換為陣列：
w = 1920
h = 1080
bmpfilenamename = r"path"
hwnd = win32gui.FindWindow(None, "my_window")
wDC = win32gui.GetWindowDC(hwnd)
dcObj=win32ui.CreateDCFromHandle(wDC)
cDC=dcObj.CreateCompatibleDC()
dataBitMap = win32ui.CreateBitmap()
dataBitMap.CreateCompatibleBitmap(dcObj, w, h)
cDC.SelectObject(dataBitMap)
cDC.BitBlt((0, 0),(w, h) , dcObj, (0, 0), win32con.SRCCOPY)
dataBitMap.SaveBitmapFile(cDC, bmpfilenamename) #i want to ignor this phase and directly convert the data to array
My_array = np.array(#myimg , dtype='float')
print(array)
print(array.shape)
我的最終目標是為神經網路預測模型提供快速的螢屏截圖流
uj5u.com熱心網友回復：
您可以使用 PIL 轉換dataBitMap為 PIL Image 物件，如此處所示。
用于array = np.asarray(im)轉換im為 NumPy 陣列。
像素格式應該是 BGRA。
這是一個代碼示例：
import win32gui
import win32ui
import win32con
import numpy as np
from PIL import Image
import cv2 # Used for showing the NumPy array as image
w = 1920
h = 1080
hwnd = win32gui.FindWindow(None, "my_window")
wDC = win32gui.GetWindowDC(hwnd)
dcObj = win32ui.CreateDCFromHandle(wDC)
cDC = dcObj.CreateCompatibleDC()
dataBitMap = win32ui.CreateBitmap()
dataBitMap.CreateCompatibleBitmap(dcObj, w, h)
cDC.SelectObject(dataBitMap)
cDC.BitBlt((0, 0), (w, h), dcObj, (0, 0), win32con.SRCCOPY)
# https://stackoverflow.com/questions/6951557/pil-and-bitmap-from-winapi
bmpinfo = dataBitMap.GetInfo()
bmpstr = dataBitMap.GetBitmapBits(True)
im = Image.frombuffer('RGBA', (bmpinfo['bmWidth'], bmpinfo['bmHeight']), bmpstr, 'raw', 'RGBA', 0, 1)
array = np.asarray(im) # Convet to NumPy array
# Show image for testing
cv2.imshow('array', array)
cv2.waitKey()
cv2.destroyAllWindows()
轉載請註明出處，本文鏈接：https://www.uj5u.com/caozuo/424379.html