2枚の画像を比較して、一致する画素を出力するプログラムを作りたい。

Question

２つの画像ファイルを比較して、どのくらい一致するのか調べるプログラムを作りたいです。画像ファイルはtiffファイルを何枚か重ねたものです。
比較として、同じ画素値の総数を出力します。
下のコードを実行すると、エラー分が出てきます。
[]をなくしてみたのですが、うまく行きませんでした。あとはどこを確認すればいいですかね？
読み込みたい画像ファイルを共有します。
https://drive.google.com/drive/folders/1zc7935SWTLych8-FN21Tm6UckCu8l5sE?usp=sharing
画像ファイルは公開されているもので、８枚のマルチスケールのtifファイルになります。

Traceback (most recent call last):
  File "/share/home/nakayama/test/PyTorch-2D-3D-UNet-Tutorial/test_image.py", line 18, in <module>
    print(np.count_nonzero(img_original == img_comparison1))
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

import cv2
import numpy as np

# 画像を比較し、一致していない箇所を表示する

# 元画像
#img_original = cv2.imread("CIMG0120_8/Input/000001_000008.tif")
r, img_original = cv2.imreadmulti('a.tif', [], cv2.IMREAD_ANYCOLOR)




# 比較対象画像１枚目(同じ画像)
#img_comparison1 = cv2.imread("CIMG0123_8/Input/000009_000016.tif")
#img_comparison1 = cv2.imreadmulti('CIMG0123_8/Input/000009_000016.tif', [], cv2.IMREAD_ANYCOLOR)
r, img_comparison1 = cv2.imreadmulti('a.tif', [], cv2.IMREAD_ANYCOLOR)

# 比較対象画像２枚目(違う画像)
r, img_comparison2 = cv2.imreadmulti('b.tif', [], cv2.IMREAD_ANYCOLOR)


# 画素がどのくらい一致しているかを確認する(一致している画素数を出力)
print(np.count_nonzero(img_original == img_comparison1))
print(np.count_nonzero(img_original == img_comparison2))

Comments

[melian]

> tiroha さん
カウントするだけで良いのであれば以下の様にすれば良いかと思います。
print(sum(np.sum(img_original[i] == img_comparison1[i]) for i in range(len(img_comparison1))))
print(sum(np.sum(img_original[i] == img_comparison2[i]) for i in range(len(img_comparison2))))

[tiroha]

ありがとうございます。
上手くいきました。この出力される数値は一致した画素値の総数であっていますか？

[melian]

はい、そうなります。np.sum を使いましたが、np.count_nonzero でも同じ結果になります。

[tiroha]

なるほど。
ありがとうございます！

Answer 1

img_original = cv2.imreadmulti('CIMG0123_8/Input/000001_000008.tif', [], cv2.IMREAD_ANYCOLOR)

が間違っています。

imreadmulti()に書かれているように、値はTupleです。

r, img_original = cv2.imreadmulti('CIMG0123_8/Input/000001_000008.tif', [], cv2.IMREAD_ANYCOLOR)

と書くのが普通でしょう。

それ以降については、思ったように動くかどうかは保証できません。
読み込むファイルがないので答えようがないからです。

Comments

[tiroha]

回答ありがとうございます。
修正したのですが、同じエラー文が出て困っています。
８枚のマルチスケールの画像ファイルを共有します。
https://drive.google.com/drive/folders/1zc7935SWTLych8-FN21Tm6UckCu8l5sE?usp=sharing

[tiroha]

回答ありがとうございました！
みなさんのおかげで、自分の作りたいプログラムを完成させることができました！

Answer 2

以下のように.all(axis=2)で色が一致するかをピクセル毎に得ることができます。

import numpy as np

# テストデータ
img1 = np.zeros((3,4,3))
img2 = np.zeros((3,4,3))
img1[:,:] = np.array([0,0,255])
img2[:,:] = np.array([0,255,0])
img1[:,1,:] = np.array([255,255,255])
img2[:,1,:] = np.array([255,255,255])

diff = (img1 == img2).all(axis=2) # BGR部分が一致
print(diff)
#[[False  True False False]
# [False  True False False]
# [False  True False False]]

ret = np.count_nonzero(diff)
print(ret) # 3