Как найти изображение в другом изображении с помощью python

Question

Я пытаюсь использовать python, чтобы определить, находится ли одно (маленькое) изображение внутри другого (большого) изображения.

Любые предложения до того, как я полностью опустил неправильный путь?

/edit: Хорошо, некоторые идеи: я использую PIL, и я преобразую каждое изображение в режим «P», чтобы сравнить каждый пиксель как целое число. Я пытаюсь реализовать что-то вроде строкового поиска Boyer-Moore или алгоритма Кнута-Морриса-Пратта, но в двух измерениях.

Может быть, это поможет: вместо того, чтобы искать ABC in XXXABCXXX (ответ = 4) мы ищем

ABC  
DEF  
GHI  

в

XXXXX   
XABCX   
XDEFX  
XGHIX  
XXXXX 

(ответ = (2,2))

Answer 1

EDIT: Хорошо, вот наивный способ сделать это:

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) 

Он работает просто отлично, но я хочу СКОРОСТИ. Я думаю, что ключ находится в массиве 'test', который мы могли бы использовать, чтобы пропустить некоторые позиции на изображении.

Редактирование 2: Убедитесь, что вы используете изображения в формате без потерь, чтобы проверить это.

На Mac установите подушку и from PIL import Image

Answer 2

Sikuli использует OpenCV, см. here, как работает match_by_template, а затем использует привязки OpenCV Python для выполнения этого же действия. Делать это без OpenCV должно быть трудно, посмотрите на OpenCV документации, поиск соответствия шаблона и т.д ...

Answer 3

Я знаю, что это немного поздно, но вы можете использовать Boyer-Moore для поиска первой линии небольшого изображения в каждой из линий большого изображения. В тот момент, когда вы находите совпадение, у вас есть позиция X и Y, и вам просто нужно проверить, совпадает ли оставшаяся часть линий меньшего изображения с остальными строками большего изображения, начиная с позиций X и Y + 1,2, 3, ... При первом несоответствии продолжаются поиск первой строки. Я не думаю, что вы можете добиться этого быстрее.

Answer 4

Посмотрите на мой ответ на аналогичный вопрос для code example using OpenCV. Преобразование из PIL в numpy является прямым, например. просто используйте np.array(pilimage).