Python OpenCVで画像を検索して判定結果をだす方法

Question

import cv2

im = cv2.imread('lena.png') # lena

im1 = cv2.imread('lena.png')
im2 = cv2.imread('who.png')

print(np.array_equal(im, im1)) # She is lena
print(np.array_equal(im, im2)) # She ain't lena!!... what?!

lena.png

who.png（lenaの切り抜き）

このプログラムだと完全に一致している場合しかTrueになりません
whoの画像をTrueにするにはどうすればいいでしょうか
よろしくお願いします

Comments

[tetsunosuke]

ndarrayの大きさがそもそも違うから、ということで、
ndarrayがもう片方に含まれるかどうかを検索する、という手もあると思いますが、
この手の方法は通常は cv2. matchTemplateで行うような気がします。その方法ではダメでしょうか？
参考：https://qiita.com/anzanshi/items/82fc4c7a3a1f84137aef

Answer 1

cv2.matchTemplateでできそうです。
詳細は以下を参照ください。
OpenCVで画像部分一致を検索したい人生だった
OpenCVを使ったテンプレートマッチング
cv::matchTemplate

import cv2
import numpy as np

img = cv2.imread("lena.png")
tpl = cv2.imread("who.png")

h, w = tpl.shape[1], tpl.shape[0]

res = cv2.matchTemplate(img, tpl, cv2.TM_CCORR_NORMED)
loc = np.where(res >= 0.99) # 閾値で絞る

if len(loc[0]) > 0:
    print('She is lena')
    pt = (loc[1][0], loc[0][0])
    cv2.rectangle(img, pt, (pt[0]+w,pt[1]+h), (255,0,0), 2)
    cv2.imwrite("res.png", img)
else:
    print("She ain't lena!!...what!?")