[AI 프로젝트] 한식 분류기(2/6)

1편에서 주제 및 dataset을 선정하는 과정에 대해서 다뤘다.

Task 수립

프로젝트를 제대로 진행하려면 무엇을 하고자 하는지 명확히 해야한다. 따라서 우리는 goal과 task를 수립했다. 우리의 궁극적인 목표는 일반적인 밥상 사진을 찍으면 그 중에서 음식을 검출하고, 검출된 음식을 분류하는 시스템을 만드는 것이다. 필요에 따라서 음식인지 아닌지를 판단하는 이진분류기가 추가될 수 있다.

하지만 한 달 남짓한 프로젝트 기간 내에 불가능하다는 것을 알고있다. 그래서 음식 분류기에 집중하기로 했다. 단일 음식 사진이 입력으로 주어지면 음식을 분류하는 음식 분류기를 개발한다.

Model 후보 사전 조사

프로젝트 동안 완전히 새로운 알고리즘을 만드는 것을 불가능이다. 기존에 발표된 image classification 중에서 성능이 좋은 것을 선택해 우리 모델에 맞게 변형하고자 한다. paperswithcode에서 뭐가 있는지 탐색했다.

후보 둘은 다음과 같다.

1. ViT
2. EfficientNet

model 후보들에 대한 자세한 내용은 다음 편에서 다루겠다.