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

연습 문제

사전 학습된 모델 불러오기

소셜 미디어에서 가져온 이미지에 레이블을 붙이는 애플리케이션을 만들고 있어요. 이 작업에는 높은 정확도와 속도가 필요합니다. 사전 학습된 ResNet18 모델을 사용해 이미지 클래스를 추론해 보세요.

지침

100 XP
  • torchvision.models에서 resnet18과 ResNet18_Weights를 임포트하세요.
  • weights를 가중치 매개변수로 설정하여 resnet18()로 model을 인스턴스화하세요.
  • model을 평가 모드로 설정하세요.
  • 입력 변환을 초기화하여 transform에 할당하세요.