파이토치로 배우는 LLM & AI

105

3

장

고급 컴퓨터 비전: 이미지에서 특징 감지하기

3.6

이미지 증식

이전 절에서는 비교적 작은 데이터셋에서 훈련해 말-사람 분류 모델을 만들었습니다. 그 결과

이전에 본 적 없는 이미지를 분류할 때 문제에 부딪혔습니다. 훈련 세트에 포함되지 않은 자세

를 취한 여성을 말이라고 잘못 분류했습니다.

이런 문제를 해결하는 한 가지 방법은 이미지 증식입니다. 이미지 증식은 파이토치가 데이터를

로드할 때 이미지에 여러 변환을 적용해 새로운 데이터를 추가로 만드는 기법입니다. 예를 들

어 [그림

3

-

11

]을 보죠. 오른쪽의 여성과 동일한 이미지는 데이터셋에 없지만 왼쪽의 이미지

가 조금 비슷합니다.

그림

3-11

데이터셋에 있는 비슷한 이미지

[그림

3

-

12

]처럼 왼쪽의 이미지를 확대해 모델을 훈련하면 오른쪽 사진 속의 사람을 올바르게

분류할 가능성이 높아집니다.

106

1

부

모델 구축

그림

3-12

훈련 세트에 있는 이미지 확대하기

비슷한 방식으로 다음과 같은 여러 가지 다양한 변환을 적용해 훈련 세트를 늘릴 수 있습니다.

•

회전

: 이미지 돌리기

•

수평 이동

: 픽셀을 수평으로 이동하고 경계를 ...