/rocopicker_django

Distinguish game charater using CNN networks

Primary LanguageJupyter Notebook

RocoPicker(로코픽커)

개요

이 프로젝트는 '아이돌 마스터 밀리언 라이브'의 캐릭터 '한다 로코'를 구분하는 인공지능을 만드는 것을 목표로 하는 토이 프로젝트입니다. 인공지능은 딥러닝 기법을 활용하여 학습시켜 두었습니다.

프로젝트 진행 과정에 대한 설명은 다음 notion페이지에서 확인할 수 있습니다.

이 github 레포지토리는 학습시킨 모델을 웹 서비스로 배포하기 위해 사용한 코드들입니다. django를 통해 작성했으며, 레포지토리 안에는 학습된 모델과 학습에 사용된 colab노트북도 같이 넣어 두었습니다.

구성

model

/model/에는 캐릭터를 구분할 수 있게 하는 학습된 모델을 업로드 해 둘 예정입니다. 이 프로젝트에서는 tensorflow 2.0 을 사용하였지만 다른 프레임워크에서 모델을 사용할 수도 있을 것이라고 생각됩니다.

또한 같은 폴더 안에 있는 Jupyter NoteBook은 google colab에서 작성한 노트북 파일이 들어 있습니다. 학습에 사용한 코드들을 볼 수 있습니다.

picker

django 를 이용해서 만든 app입니다. 임의의 사용자가 자신이 가진 이미지를 업로드 하게 하고, 업로드된 이미지가 Roco인지 판단하는 기능을 수행합니다. 샘플 이미지 두 개를 준비하여 샘플 이미지도 입력할 수 있게 할 하였습니다. 크레딧 페이지와 오류 템플릿을 포함, 6개의 뷰 템플릿이 있습니다.

picker/predictapp.py

python tensorflow 를 사용한 모델 예측 앱입니다. 현재는 Javascript 를 사용한 예측 스크립트로 예측하여 사용하지 않습니다.