我试图找到一种有效的方法,从元组(每4个条目对应一个像素的R,G,B,alpha)转换为NumPy数组(用于OpenCV).
更具体地说,我使用pywin32来获取窗口的客户区位图.这以元组的形式返回,其中前四个元素属于第一个像素的RGB-alpha通道,然后是第二个像素的下四个,依此类推.元组本身只包含整数数据(即它不包含任何维度,但我确实有这些信息).从这个元组我想创建NumPy 3D数组(宽x高x通道).目前,我只是创建一个零数组,然后遍历元组中的每个条目并将其放入NumPy数组中.我正在使用下面的代码执行此操作.而且我希望有一种更有效的方法可以做到这一点,我只是没想到.有什么建议?非常感谢!
码:
bitmapBits = dataBitmap.GetBitmapBits(False) #Gets the tuple.
clientImage = numpy.zeros((height, width, 4), numpy.uint8)
iter_channel = 0
iter_x = 0
iter_y = 0
for bit in bitmapBits:
clientImage[iter_y, iter_x, iter_channel] = bit
iter_channel += 1
if iter_channel == 4:
iter_channel = 0
iter_x += 1
if iter_x == width:
iter_x = 0
iter_y += 1
if iter_y == height:
iter_y = 0