###Python/OpenCVでエラーが出ます。
Python/OpenCVでNCCを用いた画像のテンプレートマッチングをしようとしています。
下記のコード(拾い物です)を実行しようとするとエラーが発生します。
初心者なので、よくわからず途方にくれています。
発生している問題・エラーメッセージ
OpenCV(4.1.0) /Users/travis/build/skvark/opencv-python/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'
該当のソースコード
Python
1#-*- coding:utf-8 -*- 2import cv2 3import numpy as np 4 5def template_matching_ncc(src, temp): 6 # 画像の高さ・幅を取得 7 h, w = src.shape 8 ht, wt = temp.shape 9 10 # スコア格納用の2次元リスト 11 score = np.empty((h-ht, w-wt)) 12 13 # 配列のデータ型をuint8からfloatに変換 14 src = np.array(src, dtype="float") 15 temp = np.array(temp, dtype="float") 16 17 # 走査 18 for dy in range(0, h - ht): 19 for dx in range(0, w - wt): 20 # 窓画像 21 roi = src[dy:dy + ht, dx:dx + wt] 22 # NCCの計算式 23 num = np.sum(roi * temp) 24 den = np.sqrt( (np.sum(roi ** 2))) * np.sqrt(np.sum(temp ** 2)) 25 if den == 0: score[dy, dx] = 0 26 score[dy, dx] = num / den 27 28 # スコアが最大(1に最も近い)の走査位置を返す 29 pt = np.unravel_index(score.argmax(), score.shape) 30 31 return (pt[1], pt[0]) 32 33 34def main(): 35 # 入力画像とテンプレート画像をで取得 36 img = cv2.imread("input.png") 37 temp = cv2.imread("temp.png") 38 39 # グレースケール変換 40 gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY) 41 temp = cv2.cvtColor(temp, cv2.COLOR_RGB2GRAY) 42 43 # テンプレート画像の高さ・幅 44 h, w = temp.shape 45 46 # テンプレートマッチング(NumPyで実装) 47 pt = template_matching_ncc(gray, temp) 48 49 # テンプレートマッチング(OpenCVで実装) 50 #match = cv2.matchTemplate(gray, temp, cv2.TM_CCOEFF_NORMED) 51 #min_value, max_value, min_pt, max_pt = cv2.minMaxLoc(match) 52 #pt = max_pt 53 54 # テンプレートマッチングの結果を出力 55 cv2.rectangle(img, (pt[0], pt[1] ), (pt[0] + w, pt[1] + h), (0,0,200), 3) 56 cv2.imwrite("output.png", img) 57 58 59if __name__ == "__main__": 60 main()
試したこと
どうやら、ファイルのパスの問題のようなのでいじって見ましたが変化はありませんでした。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。