본문 바로가기

Computer Vision/Classification

개와 고양이 분류하기 - 0

이번 포스팅은 개와 고양이를 분류하는 만들어 본다.

개와 고양이의 경우 미국인들이 매우매우 사랑하기 때문에.. kaggle데이터 셋만 뒤져봐도 수많은 데이터를 얻을 수 있다.

간단하게 cats vs dogs데이터 셋을 이용하여 개와 고양이를 이진 분류하고,

kaggle에서 개의 혈통 분류 데이터와 고양이 혈통 분류 데이터를 이용하여 개와 고양이의 혈통까지 알아내는 모델을 만들어 보겠다.

 

대략적인 flow chart를 그려보자면 대략 이렇다.

개와 고양이 분류 flow chart

 대략 요런식으로 flow가 진행될 것이고,

Input 을  web page에서 받아서 server를 통해 ouput을 얻은 다음 다시 web사용자에게 제공하는 형식으로 진행이 될 것이다.

가장 먼저 kaggle을 통해 cats vs dogs의 데이터 셋을 받아와 보자.

www.kaggle.com/c/dogs-vs-cats-redux-kernels-edition/data

 

Dogs vs. Cats Redux: Kernels Edition

Distinguish images of dogs from cats

www.kaggle.com

개와 고양이의 데이터는 이 홈페이지를 통해 다운로드 받을 수 있다.

나머지 혈통 데이터는 각각 dog breed, cat breed를 검색하여 다운로드 받았다.

데이터셋 준비

다운로드 받은 데이터는 총 7.6GB정도 다운로드 받았다.

위 이미지의 데이터셋은 순서대로 고양이 혈통, 고양이+개  혈통, 개 혈통, 개와 고양이 데이터 셋이다.

그리고.. 다음 포스팅에서 ..