• 목록
  • 아래로
  • 위로

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)

 

 

 

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

 

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

 

작성자
이니스프리 119 Lv. (1%) 2392650/115200000EXP

Make StudyForUs Great Again!

 

CSVpuymXAAAVVpd.jpg

댓글 1

이니스프리 작성자
profile image
https://github.com/jterrace/pyssim

참고로 Wasserstein distance와는 무관하지만 이 모듈도 괜찮은 것 같네요 ^^
comment menu
2020.07.13. 18:06

신고

"이니스프리님의 댓글"

이 댓글을 신고 하시겠습니까?

권한이 없습니다.
번호 제목 글쓴이 날짜 조회 수
공지 [작업 완료] 설 명절 맞이 서버 업데이트 안내 3 마스터 마스터 24.02.11.17:21 2205
공지 [중요] 호스팅 만료와 관련하여 일부 수칙이 변경됩니다. 4 마스터 마스터 23.01.14.02:23 5887
96 최근 KT 해외망 간단 리뷰 이니스프리 이니스프리 19.12.25.23:21 174
95 [Python] extension이 標示되지 않는 online image의 extension을 알아내기 이니스프리 이니스프리 22.04.01.20:51 174
94 [Python] COS Pro 1급 예제 2-6 풀이 이니스프리 이니스프리 20.02.14.15:03 174
93 Eva Icons - 웹, iOS, 안드로이드를 위한 오픈소스 아이콘 2 이니스프리 이니스프리 18.12.08.14:38 174
92 [Docker] Docker 다시 알고 사용하기 Seia Seia 21.04.10.07:04 173
91 앱체크 라이선스 갱신과 관련하여 이니스프리 이니스프리 19.05.11.14:43 173
90 [Python] 파이썬으로 평균을 구하는 두 가지 방법 이니스프리 이니스프리 21.03.13.21:58 171
89 [Python] Function to save list into a text file line by line 이니스프리 이니스프리 22.06.11.21:33 171
88 [Python] 독일어, 스페인어, 체코어 등 철자를 영문 알파벳으로 전환하기 이니스프리 이니스프리 24.04.21.17:15 170
87 [스퀘어넷] Ubuntu Server 18.04 LTS 지원 1 이니스프리 이니스프리 19.02.04.22:20 170
86 [Python] OrderedDict에 대한 간단한 소개 3 이니스프리 이니스프리 20.06.16.13:09 169
85 AMD KOREA 라이젠 3500X 공식 런칭 예정 image 이니스프리 이니스프리 19.10.15.17:50 169
84 [Python] 변수의 이름을 반환하는 함수 이니스프리 이니스프리 24.04.20.23:29 168
83 [펌] 하이패스 미납요금 내기 귀찮을 땐, 동네 편의점? image 이니스프리 이니스프리 19.08.28.14:53 167
82 마지막 방문일자 알려주기 슬기 슬기 18.10.11.00:31 166
81 [팁] 모바일 앱 빌더 해피보이 20.04.02.08:59 165
80 JLPT 성적표 및 합격증 수령 주소 변경 이니스프리 이니스프리 20.01.29.21:51 165
79 [Python] 구버전 KeyCaptcha 정답 좌표 찾기 image 네모 네모 21.07.09.02:46 164
78 [펌] 하나의 키 입력으로 유튜브 광고 차단하기 - Block YouTube Ads With a Single Keystroke 2 이니스프리 이니스프리 20.06.14.10:19 162
77 [PHP] 파일을 읽어서 각 행을 리스트로 만들기 이니스프리 이니스프리 19.07.10.20:29 162