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

연습 문제

Semantic segmentation 실행하기

U-Net을 잘 설계하셨습니다! 방금 만든 모델과 매우 유사한, 이미 사전 학습된 모델을 사용할 수 있도록 준비해 두었습니다. 이 모델은 대규모 이미지 데이터로 학습되었고, batch norm 레이어 같은 몇 가지 소소한 아키텍처 개선이 포함되어 있습니다.

모델은 UNet()으로 인스턴스화할 수 있으며, 사전 학습된 가중치가 포함되어 있습니다. 이제 이 모델을 사용해 아래 자동차 이미지의 segmentation mask를 생성해 보세요.

car image

PIL의 Image는 이미 임포트되어 있습니다.

지침

100 XP
  • UNet()을 model이라는 변수로 인스턴스화하세요.
  • car.jpg 이미지를 불러 image라는 변수에 저장하세요.
  • 이미지를 모델에 전달하고 출력에 squeeze(0)을 적용해 segmentation mask를 생성하세요.