[Python] 이미지에서 원하는 영역을 이진 이미지(binary mask)로 가져오기

이미지에서 원하는 위치를 찾은 후,

해당 위치를 1로 설정하고 나머지 위치를 0으로 설정하는 마스크 이미지 생성

 

np.where를 이용하여 image에서 값이 0인 위치를 찾고, 그 위치 값을 1로, 그 외를 0으로 설정한 마스크 생성

 

import numpy as np

# 주어진 이미지 (예시)
image = np.random.randint(0, 10, size=(112, 112))  # 0부터 9 사이의 랜덤 값으로 채워진 112x112 이미지

# np.where을 사용하여 0인 위치를 1로, 그 외의 위치를 0으로 설정한 마스크 이미지 생성
mask = np.where(image == 0, 1, 0)

# 마스크 이미지 확인
print(mask)