이미지에서 색상을 추출하기 위해 K-means algorithm을 구현하던 중 python library를 발견했다.
라이브러리 소개
간단히 말해 '도약 K-means algorithm'으로, 모든 픽셀을 검사하지 않고 도약할 간격을 설정할 수 있는 것이다.
Default는 10이며, 1에 가까워질 수록 생략하는 픽셀 수가 적어지므로 성능이 좋아지고 시간이 오래 걸린다.
설치 방법
pip install colorthief사용 방법
from colorthief import ColorThief
color_thief = ColorThief('/path/to/imagefile')
palette = color_thief.get_palette(color_count=6) # 여섯가지 색상 반환
# 반환 형태: [[r, g, b], [r, g, b], [r, g, b], [r, g, b], [r, g, b], [r, g, b],]
import matplotlib.pyplot as plt
plt.imshow([palette])
plt.show()이 라이브러리는 매우 가벼우면서도 효율적으로 구현되어 있다. 필요한 최소한의 종속성(math, PIL)만을 사용한다.
color_count=9를 입력했을 때 여덟 가지 색상이 반환되는 경우도 있으니 주의해야 한다.
Hi, there 👋