파이토치로 배우는 LLM & AI

102

1

부

모델 구축

모델을 훈련한 다음 ‘모델 실행하기’ 절이 나옵니다. 코드를 실행하기 전에 온라인에서 말과 사

람 이미지를 몇 개 찾아 다운로드합니다. 픽사베이(

https

://

pixabay

.

com

)는 저작권이 없는

이미지를 다운로드하기 좋은 사이트입니다. 이미지를 찾는 동안 콜랩 런타임이 종료될 수 있으

니 미리 테스트할 이미지를 준비하는 것이 좋습니다.

[그림

3

-

9

]는 픽사베이에서 모델 테스트를 위해 다운로드한 말과 사람 이미지입니다.

그림

3-9

테스트 이미지

[그림

3

-

10

]에서 보듯이 이미지를 업로드하면 모델이 한 이미지를 사람으로 다른 이미지는 말

로 올바르게 분류합니다. 하지만 첫 번째 이미지는 사람임에도 불구하고 모델이 말로 잘못 분

류합니다.

여러 개의 이미지를 업로드해 모델의 예측을 확인할 수 있습니다. 결과를 보니 말에 과대적합

된 것 같습니다. 완벽한 자세의 사람이 아니면 (즉 전신이 드러나지 않으면 ) 말로 오인될 수

있으며 이번 결과에서 이런 현상이 발생했습니다. 첫 번째 이미지는 데이터셋에 있는 다른 사

람의 이미지처럼 전신이 보이므로 사람으로 올바르게 분류합니다. 두 번째 사람은 카메라를 응

시하고 있지만 상반신만 드러나 있습니다. 훈련 데이터에는 이런 사진이 없기 때문에 모델이

올바르게 사람으로 분류하지 못합니다.

103

3

장