프로젝트/ML 웹앱 배포 전주기

ML 배포 프로젝트 개요

ryanhearts 2024. 1. 22.

프로젝트 개요

ML 서비스의 웹앱 배포 전주기 체험을 목표로 한다.

 

처음엔 반려동물(강아지/고양이)의 털 색상 추출 후 이를 통해 강아지의 펄스널 컬러 추천 서비스를 만들려 했다. 색상을 입력하면 펄스널 컬러가 딱 나오는 공식으로 쉽게 구할 수있을줄 알았기 때문. 그러나 그렇게 단순하지 않았다.. (펄스널 컬러 해본적이 없어서 몰랐음)

 

어느정도의 룰은 있지만 일반적으로 다양한 색의 천을 다 대보면서 (흡사 Brute-force..) 가장 잘 어울리는 색을 주관적으로 고르는 과정을 거쳐 구하는듯 했다. 전주기 체험이 목적인 만큼 복잡한건 잠깐 놓아두고 간단히 털 색 추출까지만 해서 구현하는걸로 결정.


프로젝트 과정

1. Segmentation model 학습 (YoloV8 + 커스텀데이터)

2. 메인 프로세스 구현 (이미지 입력 > segmentation > 색추출)

2. API 구현 (FastAPI)

3. 웹페이지 구현 (React + ChatGPT)

4. 배포 React - S3+CloudFront / API - EC2