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

연습 문제

소프트 마스크 표시하기

이전 연습 문제에서 Mask R-CNN 모델이 분할한 가장 가능성이 높은 두 객체가 모두 고양이라는 것을 확인했어요. 이제 원본 이미지 위에 이 두 고양이의 마스크를 겹쳐 표시하여 정확도를 시각적으로 확인해 보겠습니다. 이를 위해 두 개의 마스크를 순회(iterate)하면서, 각 마스크마다 원본 이미지를 먼저 표시하고 그 위에 반투명 마스크를 얹어 그리면 됩니다.

지침

100 XP
  • prediction에서 마스크와 레이블을 추출해 각각 masks와 labels에 할당하세요.
  • for 루프 안에서, mask[i, 0]을 플로팅 함수에 전달해 이미지 위에 i번째 마스크를 표시하세요. 이때 색상 맵은 "jet"을 사용하고, 투명도 매개변수는 0.5로 설정하세요.