배스킨라빈스 아이스크림 종류 중 '민트 초코'와 '엄마는 외계인'을 구분하는 모델을 간단하게 만들어보자!
1. Teachable Machine(https://teachablemachine.withgoogle.com) 사이트 접속
* Teachable Machine : 머신러닝 모델을 쉽고 빠르고 간단하게 만들 수 있도록 제작된 웹 기반 도구를 제공하는 사이트
2. 이미지 프로젝트 시작하기
[이미지 프로젝트] > [표준 이미지 모델] 클릭
3. 모델 학습시키기
분류할 종류(클래스)명 입력 > 샘플 이미지 업로드 > 모델 학습시키기
* 모델을 학습시킬 때 고급 설정으로 에포크, 배치크기, 학습률을 설정할 수 있음.
분류할 클래스명에 '민트 초코'와 '엄마는 외계인'을 입력하고
배스킨라빈스 홈페이지에서 다운로드 받은 아이스크림 이미지를 16개씩 업로드 후 모델을 학습시킴.
하이퍼 파라미터는 기본값인 에포크 50, 배치크기 16, 학습률 0.001으로 설정함.
4. 모델 테스트하기
입력-웹캠에서 파일로 변경 > 테스트 이미지 업로드
5. 결과
모델 학습에 사용하지 않은 샘플 새로운 '엄마는 외계인' 아이스크림 이미지를 모델에 입력하였고 모델은 '엄마는 외계인'으로 잘 인식함!
분류해야 하는 아이스크림 종류가 많거나 노이즈가 있는 이미지로 테스트하면 다른 결과가 나올 수도 있겠지만 '민트 초코'와 '엄마는 외계인'만을 구분하는 모델은 상당히 잘 작동하는 것 같다.
간편하게 모델을 만들어 보고 싶은 사람에게 적극 추천한다.
'민트 초코'와 '엄마는 외계인'을 구분하는 모델은 아래에서 실제로 테스트해볼 수 있다.
https://teachablemachine.withgoogle.com/models/tjdXDNRhx/
Teachable Machine
Train a computer to recognize your own images, sounds, & poses. A fast, easy way to create machine learning models for your sites, apps, and more – no expertise or coding required.
teachablemachine.withgoogle.com
'데이터 사이언스' 카테고리의 다른 글
[데이터 사이언스] Seaborn 으로 샘플 데이터셋 가져오기 (0) | 2021.04.24 |
---|---|
[데이터시각화] Seaborn Boxplot 예제 코드 설명 (0) | 2021.04.24 |
[데이터 사이언스] DW(데이터웨어하우스)와 DB(데이터베이스)의 비교 (0) | 2021.01.10 |
[데이터 사이언스] 데이터 분석 및 시각화 프로젝트 아이디어 (0) | 2020.05.22 |
[데이터 시각화] D3.js로 세계 지도에 국가별 Corona19 코로나 확진자 수 실시간으로 나타내기 (0) | 2020.05.19 |