テンプレートマッチングを行いたいです。

Question

リンク内容

import cv2
import matplotlib.pyplot as plt

# 入力画像、テンプレート画像を読み込む。
img = cv2.imread('mario.png')  # 入力画像
template = cv2.imread('template.png')  # テンプレート画像

# テンプレートマッチングを行う。
results = cv2.matchTemplate(img, template, cv2.TM_CCOEFF_NORMED)

print('img.shape', img.shape)  # (380, 694, 3)
print('template.shape', template.shape) # (30, 31, 3)

H, W, C = img.shape
h, w, c = template.shape
print(H - h + 1, W - w + 1)  # 351 664

print('result.shape', result.shape) # (351, 664)

# 検索窓の範囲を描画する。
def draw_window(img, x, y, w, h):
    '''
    Args:
        img: 描画対象の画像
        x: 検索窓の左上の x 座標
        y: 検索窓の左上の y 座標
        w: 検索窓の幅
        h: 検索窓の高さ
    '''
    tl = x, y  # 左上の頂点座標
    br = x + w, y + h  # 右下の頂点座標
    cv2.rectangle(img, tl, br, (0, 255, 0), 3)

x, y = 100, 150  # 検索窓の左上の座標
drawn = img.copy()
draw_window(drawn, x, y, w, h)

plt.imshow(cv2.cvtColor(drawn, cv2.COLOR_BGR2RGB))
plt.show()

print('similarity:', result[x, y])  # similarity: -0.0350026

# 最も類似度が高い位置を取得する。
minVal, maxVal, minLoc, maxLoc = cv2.minMaxLoc(result)
print('max value: {}, position: {}'.format(maxVal, maxLoc))
# max value: 0.9999998211860657, position: (392, 124)

# 描画する。
drawn = img.copy()
draw_window(drawn, maxLoc[0], maxLoc[1], w, h)

plt.imshow(cv2.cvtColor(drawn, cv2.COLOR_BGR2RGB))
plt.show()

# 最も類似度が 0.9 以上の位置を取得する。
locs = np.where(result >= 0.9)
drawn = img.copy()

# 描画する。
for tl in zip(*locs[::-1]):
    draw_window(drawn, tl[0], tl[1], w, h)

plt.imshow(cv2.cvtColor(drawn, cv2.COLOR_BGR2RGB))
plt.show()

リンク先の画像、プログラムをそのまま使用し実行したのですが、

img.shape (380, 694, 3)
template.shape (30, 31, 3)
351 664
result.shape (351, 664)
similarity: -0.035003584
max value: 0.9999998211860657, position: (392, 124)
Traceback (most recent call last):
  File "new14.py", line 56, in <module>
    locs = np.where(result >= 0.9)
NameError: name 'np' is not defined

このようなエラーがでてきます。
エラーの意味と改善方法を教えてください。

テラテイル自体初心者でわかりにくいと思いますが、よろしく御願い致します。

Answer 1

以下の1行を修正したら動きました。

- results = cv2.matchTemplate(img, template, cv2.TM_CCOEFF_NORMED)
+ result = cv2.matchTemplate(img, template, cv2.TM_CCOEFF_NORMED)

Comments

[stylenanda]

ありがとうございます。修正したら、動きました。
リンク先の一番下のプログラムの類似度が高い位置を描画するというプログラムも追加したら、別のエラーが発生したのですが、その場合は回答を編集してもよろしいのでしょうか

使用方法てきな質問で申し訳ありません。

[tiitoi]

現在のコードとどのようなエラーが出たかを質問を編集して記載していただけますか？

[stylenanda]

#最も類似度が0.9...からが追加したプログラムになります

[tiitoi]

> NameError: name 'np' is not defined

ファイルの一番先頭に import numpy as np
を1行追加してください

[stylenanda]

最初の構文が抜けておりました。

import したら、動きました。
初心者なので、本当にありがとうございます。