잠깐만요

이미지 분류 모델이 제대로 작동하지 않는다면?

4단계 결과확인 화면에서 Labelled Incorrectly 아래쪽에 이미지가 표시된다면 이미지 분류 모델이 테스트에 사용된 이미지 파일을 제대로 분류하지 못했다는 의미입니다.

위 그림을 보면, 5개의 키보드 이미지는 모두 키보드로 분류되었지만 5개의 리모콘 이미지 중 3개는 리모콘이 아닌 것으로 분류되었습니다. 이미지 분류 모델이 리모콘을 제대로 분류하지 못한 이유를 알아보려면 제대로 분류된 이미지와 그렇지 못한 이미지를 함께 놓고 비교해 볼 필요가 있습니다. Labelled Correctly 아래에 있는 사진 중 한 장과 Labelled Incorrectly 아래에 있는 사진 중 한 장을 선택하면 2개의 ‘Clear’ 버튼 위에 선택한 2장의 사진이 보이며, 각 사진이 어느 클래스에 속할 확률이 높은지를 알려주는 정보가 표시되어 두 사진을 한눈에 비교해 볼 수 있습니다. ‘Clear’ 버튼을 클릭하면 클릭한 버튼 위쪽에 표시되는 사진과 정보가 지워지므로 비교하는 사진을 바꾸고 싶다면 ‘Clear’ 버튼을 클릭한 후 Label Correctness 아래쪽에 있는 다른 사진을 선택하면 됩니다.

이미지 분류 모델이 리모콘을 키보드로 잘못 분류한 이유를 짐작해 봅시다. 이미지 분류 모델을 훈련시킬 때 사용한 10장의 리모콘 이미지들이 모두 흰색 배경 위에 검은색 리모콘을 세로 방향으로 놓고 촬영한 이미지입니다. 따라서 이미지 분류 모델은 이러한 공통점들을 리모콘이 갖추어야 할 조건이라고 판단하고 흰색 배경이 아닌 리모콘, 흰색 리모콘, 대각선 방향으로 놓인 리모콘은 키보드에 더 가깝다고 판단했을 것입니다.

이러한 문제를 해결하기 위해서는 다양한 모양, 색깔, 각도, 배경색을 가진 리모콘 이미지들을 추가해서 이미지 분류 모델을 다시 만들어야 합니다. 이미지 분류 모델을 다시 만들고 싶다면 화면 하단에 있는 ‘Back’ 버튼을 연속으로 클릭해서 1단계로 돌아간 후 리모콘 클래스에 새로 구한 리모콘 이미지들을 추가하고 다시 모델을 훈련시키면 됩니다. 그리고 다시 테스트했을 때 이미지 분류 모델이 여전히 이미지를 제대로 분류를 못한다면 다시 1단계로 돌아가서 새로운 이미지를 추가하고 훈련시키는 과정을 반복하면 됩니다.