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

연습 문제

사전 학습된 Mask R-CNN으로 분할하기

이 연습 문제에서는 사전 학습된 Mask R-CNN 모델을 사용해 아래 두 마리 고양이 이미지에 대해 instance segmentation을 수행해 보겠습니다.

two cats image

사용할 모델은 COCO dataset으로 사전 학습되어 있으며, 동물을 포함한 일상적인 사물 이미지가 들어 있습니다. 덕분에 별도 미세 조정 없이도 모델이 기본으로 고양이를 인식할 수 있어요.

여러분의 할 일은 모델과 두 마리 고양이 이미지를 불러오고, 이미지를 전처리한 뒤 모델에 전달해 예측 결과를 얻는 것입니다. PIL의 Image, torch, torchvision의 transforms, 그리고 maskrcnn_resnet50_fpn 은 이미 임포트되어 있습니다.

지침

100 XP
  • maskrcnn_resnet50_fpn() 을 사용해 pretrained Mask R-CNN을 model 에 로드하세요.
  • 두 마리 고양이 이미지를 텐서로 변환한 뒤 unsqueeze 하세요.
  • 이미지를 모델에 전달해 추론을 수행하고, 출력을 prediction 에 할당하세요.