[DL] 딥러닝 토이 프로젝트 (BTS 외모 순위)
Jun ・ 2020. 4. 20. 19:12
Intro
오늘은 외모순위를 알려주는 인공지능 프로그램을 만들어보려고 한다. 본 프로그램은 지도학습으로 짜였기 때문에 재미로만 보는것이지 이게 답이라고 할 수 없으니 어떤 결과가 나오든 BTS 팬 분들 살려주세요.
본 모델에서 학습한 데이터셋은 대략 아래와 같다. 추가적인 데이터셋 설명을 원하시는 분은 제일 아래 레퍼런스 사이트를 참고.
여러분은 최종적으로 BTS 멤버들의 외모 점수를 확인할 수 있을 것이다.
( 정국 - 뷔 - 진/RM - 제이홉 - 지민 - 슈가 )
dataset : https://drive.google.com/open?id=1w0TorBfTIqbquQVd6k3h_77ypnrvfGwf
Content
1. 전처리
- 이미지 처리, 시각화, 수치계산, 딥러닝 모델 프레임워크 라이브러리를 불러온다.
- 각 이미지의 레이블 값을 불러온다. 레이블은 1~5점 사이의 외모 점수이다.
- 이미지를 불러와 전처리한 후 레이블과 묶어서 numpy 형태로 변환한다.
- train data 4950개와 validation data 550개로 이루어져 있다.
2. 학습
- 위에서 numpy로 변환한 데이터를 불러온다.
- 10개의 이미지를 불러와서 보면 레이블과 이미지가 매칭된것을 확인할 수 있다.
- 이미지의 수가 부족함으로 data generator를 사용하여 데이터를 늘려준다.
- 학습.. 시간이 오래걸리므로 gpu를 사용하거나 필자 git에 있는 h5파일을 바로 사용하자
3. 평가
- 평가 데이터를 찍어본다. 결과값의 왼쪽이 예측값이고 오른쪽이 원래 값이다.
4. 테스트
- 테스트로 사용할 BTS분들의 이미지를 불러온다.
