사전 학습된 모델 불러오기
소셜 미디어에서 가져온 이미지에 레이블을 붙이는 애플리케이션을 만들고 있어요. 이 작업에는 높은 정확도와 속도가 필요합니다. 사전 학습된 ResNet18 모델을 사용해 이미지 클래스를 추론해 보세요.
이 연습은 강의의 일부입니다
PyTorch로 배우는 이미지 딥러닝
연습 안내
torchvision.models에서resnet18과ResNet18_Weights를 임포트하세요.weights를 가중치 매개변수로 설정하여resnet18()로model을 인스턴스화하세요.model을 평가 모드로 설정하세요.- 입력 변환을 초기화하여
transform에 할당하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Import resnet18 model
from torchvision.models import ____
# Initialize model with default weights
weights = ResNet18_Weights.DEFAULT
model = ____
# Set model to evaluation mode
model.____
# Initialize the transforms
transform = ____