이미지 내 색상 별 픽셀 수 확인하기

어떤 이미지 안에 object 별로 annotation이 된 segmentation mask를 받았는데,

어떤 색상들로 구성되어 있는지 모를때.. 확인이 필요했다.

 

예를 들면, Pascal VOC 데이터셋 같은 경우는 색상별 어떤 클래스인지 나와있지만

그렇지 않은 경우에는 클래스가 몇개가 있고, 어떤 색상으로 되어있는지 확인이 필요하다.

 

간단하게 확인할 수 있는데,

from PIL import Image
import matplotlib.pyplot as plt

path = '/data/2030/BileDuct/TestImage/mask/10.png'

# 이미지 열기
image = Image.open(path)
plt.imshow(image)

segmentation mask 예시

(위 이미지는 간 수술 영상에서 담관을 annotation한 mask 입니다)

# 색상 모드 확인
print(image.mode)

# 이미지 색상 값 확인
colors = image.getcolors()
print(colors)

RGB
[(53262, (128, 0, 0)), (6099, (0, 128, 0)), (2014239, (0, 0, 0))]

 

결과를 보면 빨간색 픽셀이 53,262개, 초록색이 6,099개, 검정색이 2,014,239개 있는걸 확인할 수 있다!

 

끝!