인물 사진에 특화된 이미지 생성 AI 학습 도구 나왔다

이미지 생성 AI인 FLUX 개발팀이 고품질 포트레이트 생성에 특화된 새로운 트레이너인 FLUX 포트레이트 트레이너(FLUX Portrait Trainer)를 공개했다. 특징으로는 섬세한 디테일, 눈동자 하이라이트, 다양한 얼굴 크기에 대한 더 나은 표현 등이 언급되고 있다.

샘플로 공개된 FLUX 포트레이트 트레이너 동일 모델을 사용해 생성된 이미지를 보면 체모 한 가닥 한 가닥과 눈동자에 비친 하이라이트, 피부의 질감 등 섬세한 디테일이 높은 품질로 표현되어 있어 개발팀은 그림이 화면에서 튀어나오는 듯 보인다고 말했다.

이 이미지는 CheeseAI가 제공한 데이터세트를 바탕으로 프롬프트(kptz woman in a pitch black room. I single shaft of volumetric light slants across her right eye.)를 넣어 생성한 이미지다. 완전히 어두운 방(pitch black room)이라는 지시가 포함되어 있지만 데이터세트에는 낮에 촬영한 사진만 포함되어 있었다고 한다. 왼쪽은 FLUX 포트레이트 트레이너로 트레이닝된 모델에 의한 생성으로 의도한 대로의 이미지를 출력할 수 있었지만 오른쪽 범용 트레이너로 트레이닝된 모델에 의한 생성은 완전히 어두운 상태를 만들 수 없었다.

마찬가지로 다른 프롬프트(kptz old woman)로 동일 시드·동일 파라미터 조건 하에 생성된 이미지를 보면 왼쪽 FLUX 포트레이트 트레이너로 트레이닝된 모델에 의한 생성 이미지는 프롬프트에서 요구한 대로의 이미지가 됐지만 오른쪽 범용 트레이너로 트레이닝된 모델에 의한 생성 이미지는 트레이닝에 사용된 데이터와 일치하는 이미지가 출력되어 프롬프트가 무시됐다.

한편 FLUX 포트레이트 트레이너는 범용 트레이너를 보완하는 것으로 애플리케이션에 따라서는 다른 트레이너가 더 잘 작동할 수 있다고 개발팀은 설명하고 있다. 관련 내용은 이곳에서 확인할 수 있다.