이진 마스크 생성하기
세그멘테이션 과제의 이미지는 보통 픽셀 단위 마스크로 주석 처리됩니다. 아래는 Egyptian Mau 고양이 이미지예요.
이번 연습 문제와 다음 연습 문제에서, 해당 마스크를 사용해 이미지에서 고양이만 분리해 보겠습니다. 먼저 마스크를 불러오고 이진화해야 해요.
PIL의 Image, torchvision의 transforms, 그리고 torch는 이미 임포트되어 있어요.
이 연습은 강의의 일부입니다
PyTorch로 배우는 이미지 딥러닝
연습 안내
annotations/Egyptian_Mau_123.png에 저장된 마스크 이미지를 불러와mask에 할당하세요.mask_tensor에서 각 픽셀이1/255와 같은 경우 텐서 값1.0을, 나머지 픽셀에는 텐서 값0.0을 할당해binary_mask를 만드세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Load mask image
mask = ____
# Transform mask to tensor
transform = transforms.Compose([transforms.ToTensor()])
mask_tensor = transform(mask)
# Create binary mask
binary_mask = ____(
____,
____,
____,
)
# Print unique mask values
print(binary_mask.unique())