생성형 AI를 활용한 텍스트 기반 로고 생성 모델

기간: 2022.08 - 2023.02

참여도: 100 %

프로젝트 개요

텍스트와 기업 성격을 입력하면 이를 반영한 맞춤형 로고 이미지를 생성하는 프로젝트입니다.

사전학습된 Stable Diffusion을 기반으로, 브랜드의 성격을 시각적으로 표현하는 데 중점을 두었습니다.

주요 기능

1. ‘진실된’, ‘세련된’ 등 기업성격에 어울리는 이미지 스타일을 반영해 로고를 생성합니다.
2. Stable Diffusion을 파인튜닝하여 일반적인 텍스트 입력에도 강건한 로고 생성이 가능합니다.

기술 스택

데이터 수집: Python, Selenium

모델링: Python, Pytorch, Diffusers

데이터 전처리: OpenCV

개발 내용

기업 성격 - 기업 로고 매칭 알고리즘

로고 이미지의 원형성, 대칭성 등 디자인 요소를 분석해, 기업의 성격에 자동으로 매칭하는 알고리즘을 개발했습니다.

현실적으로 비효율적인 수작업을 대신해, 팀 기준으로 200시간 이상 절약 효과를 확인했습니다.

사전학습된 Stable Diffusion 모델 파인튜닝

기존 로고 생성 모델은 클래스 레이블에 의존한 방식이 대부분이었습니다.

일반화 성능을 강화하기 위해 기업 성격/스타일 기반 데이터셋으로 사전학습된 모델을 파인튜닝했습니다.

코드 상세

데이터셋 전처리

로고 이미지 프로세스를 통해서 디자인 요소와 기업의 성격을 매칭합니다.

vscode 환경에서 가상환경을 만들어 코드를 동작하는 것을 추천합니다.

필요한 패키지 설치 후 아래 코드로 기업의 성격을 추출할 수 있습니다.

pip install -r requirements.txt
python main.py 

main.py에서 실행할 이미지 폴더를 수정하는 방식으로 실행 가능합니다.

stable-diffusion finetutning

이미지 생성을 위한 모델 파인튜닝이 필요합니다.

이 경우 아래의 diffusers github에서 pretrained 모델을 파인튜닝할 수 있습니다.

https://github.com/huggingface/diffusers