이번 포스팅에서는 스테이블 디퓨전으로 이미지를 생성할 때 사용하는 각종 옵션의 설정을 설명하겠습니다. 적정값을 설정하지 않으면, 이미지가 제대로 생성되지 않는 경우가 있습니다. 옵션 항목에는 sampling method(이미지 생성 알고리즘), 이미지의 크기, 동시에 생성하는 이미지의 숫자, seed 번호 등이 있습니다.
기본적인 화면 UI
우선 이미지 생성 포스팅에서 설명했던 Checkpoint 선택 창이 화면 상단에 있습니다. 새로 Checkpoint를 적용했다면, 우측의 Refresh 버튼을 눌러 목록을 갱신하고 선택합니다. 바로 밑에 텍스트 프롬프트, 네거티브 프롬프트 입력창이 있습니다. 그리고 그 밑으로 이번 포스팅에서 설명할 옵션 항목이 있습니다.
옵션의 설정
옵션 상단의 탭에서 Checkpoint와 Lora를 적용할 수 있습니다. Lora는 프롬프트 작성 중에 선택하여, 프롬프트 형식으로 반영한다는 것을 기억하시기를 바랍니다.
Sampling method
Sampling method는 스테이블 디퓨전이 이미지를 생성하는 과정에서 사용하는 알고리즘을 의미합니다. 선택한 Sampling method는 이미지의 다양성, 품질 및 생성 속도에 영향을 미칩니다. 일반적으로 2M Karras, Euler A, SDE Karras를 많이 사용합니다. 직접 사용하시면서 본인의 마음에 드는 Sampling method를 선택하시기를 바랍니다.
Sampling steps
Sampling steps는 이미지를 생성하는 과정에서 수행하는 작업의 횟수를 말합니다. steps가 높을수록 이미지의 세부 품질이 좋아지지만 이미지의 생성 속도가 느려집니다. 더불어 수치를 너무 높이면 이미지가 깨지는 경우가 있습니다. 개인적인 경험으로는 20~30을 적절한 수치로 생각합니다.
Width & Height
이미지의 가로 x 세로 비율을 나타냅니다. 처음부터 고해상도의 이미지를 생성하면, 많은 이미지를 생성하기 어렵고 속도도 느립니다. 개인적인 경험으로는 저해상도의 이미지를 생성하면서 옵션 및 프롬프트를 수정하는 것을 추천합니다. 어느 정도 원하는 이미지가 생성되면, 업스케일링을 적용하여 고해상도의 이미지를 생성합니다. 다만 본인의 PC 사양이 높아서 로컬로 이미지를 생성하고 있다면, 이미지의 생성 속도는 큰 문제가 아닐 수 있습니다. 이 부분은 본인의 PC 사양, 로컬/코랩 사용 여부에 따라 판단하시기를 바랍니다.
기본적으로 많이 사용하는 해상도는 512 x 512, 512 x 768, 768 x 512입니다. 업스케일링은 나중에 별도의 포스팅으로 설명하겠습니다.
Batch count & Batch size
Batch count는 차례대로 생성하는 이미지의 숫자를 의미합니다. 반면 Batch size는 동시에 생성하는 이미지의 숫자를 말합니다. Batch size는 본인의 GPU 성능에 영향을 많이 받습니다. 한 번에 많은 수의 이미지를 생성하고 싶다면, Batch count 조정을 추천합니다. 물론 본인의 PC 사양이 높다면 필요에 따라 두 옵션 모두 조정할 수 있습니다.
CFG Scale
이미지가 프롬프트를 얼마나 충실히 반영하는지 나타내는 척도입니다. 수치가 높으면 프롬프트를 그대로 반영하는 이미지가 생성됩니다. 반대로 수치가 낮으면 프롬프트와 별개로, 창의적인 이미지가 생성됩니다. 보통 7~9를 많이 사용하지만, 이미지에 변형을 주고 싶다면 낮은 수치를 사용해도 무방합니다. 이미지를 생성하시면서, 취향껏 수치를 조정하시기를 바랍니다.
Seed 번호
Seed는 이미지를 생성할 때 부여되는 고유 번호입니다. 기존에 생성한 이미지를 활용하여 새로운 이미지를 생성하고 싶다면 Seed 번호를 활용하시면 됩니다. 이미지 생성창 하단의 번호를 복사하거나, Seed 번호 입력창 우측의 녹색 버튼을 누르면 Seed 번호가 삽입됩니다.
정리
스테이블 디퓨전은 각종 옵션을 변경할 때마다, 이미지가 크게 변하는 특징이 있습니다. 심지어 이미지의 크기 변경에도 반응합니다. 여기에 Checkpoint, Lora까지 반영한다면, 원하는 이미지를 생성하는 것은 생각보다 쉽지 않습니다. 시행착오를 겪으면서 조정하는 수 밖에 없습니다. 다양하게 옵션을 조정하면서 원하는 이미지를 생성해 보시기를 바랍니다.