다음 그림은 24개의 이미지를 출력한 결과입니다.

▲ 그림 5-35 24개의 이미지 출력 결과

데이터가 준비되었으므로 사전 훈련된 ResNet18 모델을 내려받습니다.

코드 5-15 사전 훈련된 모델 내려받기

resnet18 = models.resnet18(pretrained=True) ------ pretrained=True는 사전 학습된 가중치를 사용하겠다는 의미