[Scipy] Wasserstein distance를 이용한 두 이미지 간 유사도 측정

https://stackoverflow.com/a/49592509 여기 올라온 아이디어를 참고하였습니다 ^^

(현재의 scipy 버전에 맞게 살짝 수정했네요.)

Wasserstein distance에 대해서는 아래 문서를 참고하세요!

https://kionkim.github.io/2018/06/01/WGAN_1/

pilmode에 대해서는 아래 문서를 참고하세요 :)

https://imageio.readthedocs.io/en/stable/format_bmp-pil.html?highlight=pilmode#parameters-for-reading

from scipy.stats import wasserstein_distance
from imageio import imread
import numpy as np

def get_histogram(img):
  h, w = img.shape
  hist = [0.0] * 256
  for i in range(h):
    for j in range(w):
      hist[img[i, j]] += 1
  return np.array(hist) / (h * w)

a = imread('file01.jpg', pilmode='L')
b = imread('file02.jpg', pilmode='L')
a_hist = get_histogram(a)
b_hist = get_histogram(b)
dist = wasserstein_distance(a_hist, b_hist)
print(dist)

제가 테스트해보니 칼라 이미지에서는 아주 좋은 결과를 얻기는 어려운 것 같네요 ㅠㅠ

그럼 즐거운 불금 되세요! ^-^