opencv画像検索-python

Question

Aの画像からBの画像を複数個検索したい

Aの画像の中にBの画像が複数箇所あるので全ての位置を取得したいです。
画像A

画像B

今の状態
画像を検索することはできたのですが、1つだけだったので、これを複数個検索できるようにしたいです。

import cv2
def Judge_Matching(num):
    if 0.99 < num:
        return True
    else:
        return False

# 検索元／先画像を読み込み
image = cv2.imread(r"画像A") #検索先画像のファイルパスを指定
template = cv2.imread(r"画像B") #検索元画像ファイルパスを指定

#OpenCVで画像部分一致を検索
result = cv2.matchTemplate(image, template, cv2.TM_CCOEFF)

# 最も類似度が高い位置と低い位置を取得します
minVal, maxVal, minLoc, maxLoc = cv2.minMaxLoc(result)

#類似度が閾値を超えているか判定（上で作った関数を使用）
Judg = Judge_Matching(maxVal)

x,y=maxLoc
print(x,y)
print(maxVal)
#結果を出力
print(Judg)

Answer 1

複数物体のテンプレートマッチング
を少し修正してみました。

import cv2
import numpy as np

img_rgb = cv2.imread('all.png')
img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)
template = cv2.imread('ptn.png',0)
w, h = template.shape[::-1]

res = cv2.matchTemplate(img_gray,template,cv2.TM_CCOEFF_NORMED)
threshold = 0.88
loc = np.where( res >= threshold)
for pt in zip(*loc[::-1]):
    _ = cv2.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0,0,255), 2)

cv2.imwrite('res.png',img_rgb)

実行結果

Comments

[sibainu]

座標の取得の仕方も欲しいです。
欲張ってすみません。

[ppaul]

回答のコードのlocに入っています。
ただし、一つ目の図は少しずらしてもマッチするようでほぼ同じのが3つか4つ入っていました。
ほぼ同じ座標のものを一つにするのはそれほど難しくないので自力でやっｔみてください。

[sibainu]

了解です。ありがとうございます。