python - 在另一个图像中查找透明(png)图像

Question

我有一个屏幕截图，我需要在上面找到一些图像(始终具有相同的大小、旋转角度等)。我找到了一些 PIL 和 numpy 的解决方案，但它们只适用于非透明图像。我必须找到类似圆圈的东西，所以我必须在它后面使用透明背景。

示例图像如下所示:

我正在寻找这样的目标:

有什么想法可以实现吗？

Answer 1

由于您要精确匹配图像，因此很容易创建一个 slider 来找到目标。这不是最快的解决方案，但很容易开始工作。

import numpy as np
from scipy.misc import imread

screen = imread("screen.png")
target = imread("cookie.png")

def find_match(screen, target):
    dx,dy,_ = target.shape

    for x in xrange(screen.shape[0]-dx):
        for y in xrange(screen.shape[1]-dy):
            diff = (screen[x:x+dx,y:y+dy,:]-target)
            dz = np.abs(diff).sum()
            if dz == 0: return x,y, dx, dy
    return None, None, None, None

x,y,dx,dy = find_match(screen, target)

# Show the result

import pylab as plt
plt.subplot(121)
plt.imshow(screen)

screen_sample = np.copy(screen)
screen_sample[x:x+dx,y:y+dy,:] = 0
plt.subplot(122)
plt.imshow(screen_sample)

plt.tight_layout()
plt.show()