如何使用python在另一个图像中查找图像

Z时代
2024-01-10
分类：问答

我正在尝试使用python确定一个（小）图像是否在另一个（大）图像中。

在我完全走错路之前有什么建议吗？

/ edit：好的，有些想法：我正在使用PIL，并将每个图像转换为“ P”模式，以便可以将每个像素比较为整数。我正在尝试实现诸如Boyer-

Moore字符串搜索或Knuth-Morris-Pratt算法之类的东西，但是要实现2维。

也许会有所帮助：我们不是在搜索ABC in XXXABCXXX（answer = 4），而是在搜索

ABC DEF GHI

在

XXXXX XABCX XDEFX XGHIX XXXXX

（answer =（2,2））

回答：

编辑：好的，这是天真的方法：

import Image, numpy
def subimg(img1,img2):
    img1=numpy.asarray(img1)
    img2=numpy.asarray(img2)
    #img1=numpy.array([[1,2,3],[4,5,6],[7,8,9]])
    #img2=numpy.array([[0,0,0,0,0],[0,1,2,3,0],[0,4,5,6,0],[0,7,8,9,0],[0,0,0,0,0]])
    img1y=img1.shape[0]
    img1x=img1.shape[1]
    img2y=img2.shape[0]
    img2x=img2.shape[1]
    stopy=img2y-img1y+1
    stopx=img2x-img1x+1
    for x1 in range(0,stopx):
        for y1 in range(0,stopy):
            x2=x1+img1x
            y2=y1+img1y
            pic=img2[y1:y2,x1:x2]
            test=pic==img1
            if test.all():
                return x1, y1
    return False
small=Image.open('small.tif')
big=Image.open('big.tif')
print subimg(small, big)