1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch로 배우는 이미지 딥러닝
Connected

연습 문제

이진 마스크 생성하기

세그멘테이션 과제의 이미지는 보통 픽셀 단위 마스크로 주석 처리됩니다. 아래는 Egyptian Mau 고양이 이미지예요.

cat image

이번 연습 문제와 다음 연습 문제에서, 해당 마스크를 사용해 이미지에서 고양이만 분리해 보겠습니다. 먼저 마스크를 불러오고 이진화해야 해요.

PIL의 Image, torchvision의 transforms, 그리고 torch는 이미 임포트되어 있어요.

지침

100 XP
  • annotations/Egyptian_Mau_123.png에 저장된 마스크 이미지를 불러와 mask에 할당하세요.
  • mask_tensor에서 각 픽셀이 1/255와 같은 경우 텐서 값 1.0을, 나머지 픽셀에는 텐서 값 0.0을 할당해 binary_mask를 만드세요.