Python 이미지에서 주요 색상 추출하기 (colorthief)

Python 이미지에서 주요 색상 추출하기 (colorthief)

Extracting Dominant Colors From Images in Python

 

Color Thief란 Color Thief 패키지에는 다양한 환경과 빌드 프로세스를 지원하는 여러 배포 파일이 포함되어 있습니다. 이미지에서 색상 팔레트를 가져오기 위한 모듈입니다.

 

결과

testimg.png

이미지에 존재하는 모든 색, 특정 개수의 색들을 추출할 수 있습니다. 지금의 결과는 이미지에서 5개의 주요 색상을 추출하는 라인을 실행시켰을 때의 결과입니다. 

 

라이브러리 설치

pip install colorthief

라이브러리 import

from colorthief import ColorThief
import matplotlib.pyplot as plt
import colorsys

 

이미지의 주요 색깔 추출

ct = ColorThief("testimg5.jpg")
dominant_color = ct.get_color(quality=1)
 
plt.imshow([[dominant_color]])
plt.show()

dominant color

testimg의 주요 색깔을 추출했습니다. 즉, 이미지에서 가장 많은 비율을 차지하는 색깔이 추출된 색깔과 같습니다.

 

 

이미지 내 색깔 팔레트 추출하기

palette = ct.get_palette(color_count=5)
plt.imshow([[palette[i] for i in range(5)]])
plt.show()

이미지 내에 등장하는 모든 색을 팔레트 형식으로 볼 수 있습니다. 더 자세한 내용은 아래 color thief 모듈 다큐멘테이션에서 확인할 수 있습니다.

 

 

https://lokeshdhakar.com/projects/color-thief/