헤어스타일 변경 생성 AI 네트워크 (HSTNet)
Hair Style Transfer Solution
자체개발 인공지능 모델 HSTNet(Hair Style Transfer Network)을 통해 사용자가 원하는 헤어스타일에 기반한 고수준 실사형 이미지를 제공합니다.
적용 산업 분야
Fashion
기반 모델
Stable Diffusion
솔루션 개요
동일 인물의 Bald 이미지와 Hair Referernce 이미지로부터 헤어스타일이 전이된 이미지를 생성하는 인공지능 모델
주요 특징
U-Net과 Diffusion Model을 결합한 모델 아키텍처 구현
ControlNet 기법을 접목하여 얼굴 특성 일관성 유지
Bald 상태 이미지를 생성하기 위한 Bald Image Converter 자체 구현
핵심 원리
Original Img를 자체개발한 Converter에 입력하여 Bald 이미지를 출력
Bald 이미지를 ControlNet 기반 Latent IdentityNet에 입력하여 얼굴에 대한 특성 정보를 추출
헤어스타일 Reference 이미지를 U-Net에 입력하여 머리카락에 대한 정보를 추출
2와 3에서 인코딩 된 정보를 Stable Diffusion 기반 모델에 입력하여 최종 헤어스타일이 반영된 이미지를 생성
PSNR
0.690
SSIM
Generation Time