커맨드로 쉽게 이미지를 생성하는「ImaginAIry」

이미지 생성 AI「Stable Diffusion」은 문장(프롬프트)을 입력하는 것 만으로 좋아하는 이미지를 생성해 줍니다만, 로컬 환경에서 Stable Diffusion을 사용하는 경우는 복잡한 초기 설정이 필요합니다. 오픈 소스로 개발되어 있는 Python 패키지「ImaginAIry」라면, Stable Diffusion을 이용한 이미지 생성 환경을 pip로 간단하게 구축 가능하고, 이미지 생성 이외에도 프롬프트에 의한 이미지 편집이나 초해상 기능도 이용할 수 있어, 아주 편리할 것 같아서 그 기능을 정리해 보았습니다.

GitHub - brycedrennan/imaginAIry: AI imagined images. Pythonic generation of stable diffusion images.

GitHub - brycedrennan/imaginAIry: AI imagined images. Pythonic generation of stable diffusion images.

ImaginAIry는「Linux」와「macOS(M1)」에 대응한 Python 스크립트로, 아래의 명령을 실행하면 즉시 설치할 수 있습니다.

"just works" on Linux and macOS(M1) (and maybe windows?).

pip install imaginairy

ImaginAIry를 설치하면 나중에 「imagine "프롬프트"」라는 형식의 명령을 실행하여 이미지를 생성할 수 있습니다. 프롬프트는 여러 번 동시에 지정할 수 있으며, 예를 들어, 다음 명령은 "매우 좋은 풍경(a scenic landscape)", "강아지 사진 (a photo of a dog)", "과일 그릇 사진 (photo of a fruit bowl)", "주근깨 여성의 인물사진(portrait photo of a freckled woman)"이라는 4개의 프롬프트를 지정하고 있습니다.

imagine "a scenic landscape" "a photo of a dog" "photo of a fruit bowl" "portrait photo of a freckled woman"

상기의 커맨드로 생성된 4장의 이미지가 아래. 매우 간단하게 Stable Diffusion으로 이미지 생성을 수행할 수 있습니다.

또한 ImaginAIry에는 프롬프트에 의해 이미지 편집을 실행할 수 있는 "InstructPix2Pix" 기능이 내장되어 있으며, "aimg edit 파일명 "프롬프트" --prompt-strength 편집 강도를 나타내는 수치" 형식의 명령을 실행하기만 하면 이미지 파일을 편집할 수 있습니다.

예를 들어, 앞에서 설명한 커맨드로 생성한 개의 사진(dog.jpg)에 대해「개를 적색으로 한다」라는 편집을 하고 싶은 경우는 아래의 커맨드를 실행하면 OK.

aimg edit dog.jpg "make the dog red" --prompt-strength 5

편집 전의 이미지(왼쪽)와 편집 후의 이미지(오른쪽)를 비교하면 이런 느낌. 프롬프트에서 지정한 대로 개 부분만 빨간색입니다.

또한 ImaginAIry에는 "CodeFormer"를 이용하여 "사람의 얼굴을 명료하게 하는 기능"도 탑재되어 있습니다. 얼굴을 명료하게 하는 명령의 예는 아래와 같습니다.

imagine "a couple smiling" --steps 40 --seed 1 --fix-faces

아래는 위 커맨드에 의한 편집 결과의 비교입니다.

ImaginAIry에서는, 그 밖에도 이미지를 입력해 비슷한 이미지를 출력하는「img2img」나, 이미지를 타일 형상으로 늘어놓는 기능도 탑재되어 있습니다.