영리의 테크블로그

AI 이미지 생성 프로젝트 ( AI Seller ) - [2] / ComfyUI 본문

dev/AISeller

AI 이미지 생성 프로젝트 ( AI Seller ) - [2] / ComfyUI

영리0 2024. 10. 10. 02:25

메인 프로젝트를 진행하면서 사이드 프로젝트 + 일기 까지 쓰려니 시간이 매우매우매우 부족하다...

 

프로젝트 GitHub

https://github.com/YoonJae00/AI-IMGREMOVE

 


 드디어 사이트 초안을 배포했다. 대량의 이미지를 AI로 자동 처리하는 프로그램인데, 처음 시작할 때부터 기대를 많이 했던 프로젝트다. 뭔가 설레면서도 걱정이 되는 그 느낌.

사이트 초안



배포하고 나서 방문자 유입량이 궁금해서, 간단하게 통계 사이트도 만들어봤다. 솔직히 얼마나 들어올지 모르겠지만, 그래도 뭔가 기록이 남으니까 재미있다. 

로그1
로그2



또한 자꾸 내가 IP 제한 에 걸려 이를 해결할 로직을 추가했다. 하루에 한 IP에서 10번 이상 요청이 안 되도록 설정했는데, 예외적으로 localhost는 무제한으로 접근할 수 있도록 해놨다.

```javascript
// 무제한 IP 목록 정의
const unlimitedIPs = ['127.0.0.1'];
...
```



이제 중요한 피드백이다. 배포하고 나서 바로 피드백을 받았는데, 예상했던 대로 몇 가지 문제가 있었다. 우선, 이미지 처리 시간이 좀 길다는 지적을 받았고, 배경이 복잡한 경우에선 제대로 제거가 안 되는 경우도 있었다. 피드백이 솔직히 아프긴 했지만, 개선할 부분이 뚜렷해져서 오히려 좋다.

피드백



문제를 정리하자면:
1. 퀄리티 문제
2. 시간 문제
3. UI/UX 문제

퀄리티와 시간 문제는 역시 서로 비례하는 것 같다. 퀄리티를 높이면 시간이 오래 걸리고, 시간을 줄이면 퀄리티가 떨어지고… 딜레마에 빠져 있었는데, 구글링을 한참 하다 보니 최신 모델 중에 빠르고 퀄리티도 나쁘지 않은 모델을 찾았다! 적용 전후 비교해봤는데, 꽤 괜찮은 결과를 얻을 수 있었다.

기존 15.69초 걸렸던 모델

 

개선 후 무려 1.89초!!!

 

 

개선 전 이미지 배경 제거
개선 후 이미지 배경 제거

 

 

시간도 10배 이상 빠르고 효과도 좋은 모델을 찾았다니!!!!

아주 기분이 좋다

그리고 이미지 전체 삭제 기능도 추가했다. 사용자 입장에서 한 번에 정리할 수 있는 기능이 필요한 것 같아서 넣어봤다.