[개발][유용한 툴] Stable Diffusion AI 사용법

 

# Stable Diffusion AI 사용 방법

-  최근 AI를 사용해서 그린 그림들의 결과물이 엄청나게 좋아지면서, 이에 대한 글이나 영상을 많이 보셨을 것 같습니다. 본 포스팅에서는 Stable Diffusion AI를 사용해서 이미지를 만드는 과정을 소개하려고 합니다. 개발을 하시는 분들이라면, 금방 사용하실 수 있을 것 같습니다.

# 사용 방법

- 해당 소스코드는 Stable Diffusion를 위한 Gradio라는 라이브러리를 호출하기 위한 Web UI인데, 이를 로컬에서 받아서 사용할 수 있습니다. 저는 이 소스코드를 사용하였고, Window 환경에서 진행 하였습니다.

 

 

 

GitHub - AUTOMATIC1111/stable-diffusion-webui: Stable Diffusion web UI

 

- 위의 소스코드를 실행하기 위해서는 Python 3.10.6 과 Git이 필요합니다. 아래의 링크나 명령어로 설치를 진행하시면 됩니다. 저는 Git은 이미 설치가 되어 있었고, Python의 경우 버젼 차이로 다시 설치하였습니다.

• Python 3.10.6 (webpage, exe, or win7 version)
• Git (webpage)

- 이제 소스 코드를 다운 받아줍니다. Git Clone을 하셔도 되고, 직접 다운 받으셔도 됩니다.

- 이제 학습 모델 파일을 다운 받아야 합니다. 소스코드의 설명에는 아래의 세 링크에서 받을 수 있다고 되어 있습니다. 저의 경우, 공식 다운로드는 로그인이 필요하고, 토렌트는 사용하지 않아서 파일 스토리지에서 다운로드 받았습니다. 참고하시기 바랍니다. (sd-v1-4.ckpt 파일 하나를 받으시면 됩니다. 용량이 크기 때문에 시간이 오래 걸립니다)

• Official download
• File storage
• Torrent (magnet:?xt=urn:btih:3a4a612d75ed088ea542acac52f9f45987488d1c&dn=sd-v1-4.ckpt&tr=udp%3a%2f%2ftracker.openbittorrent.com%3a6969%2fannounce&tr=udp%3a%2f%2ftracker.opentrackr.org%3a1337)

- 이제 받은 .ckpt파일을 다운 받은 소스코드의 하위 경로인 models/Stable-diffusion 폴더 안에 넣어줍니다. 그리고 webui-user.bat 파일을 실행합니다. 앞의 과정이 정상적으로 진행 되었다면, 필요한 추가 요소들을 자동으로 설치하고, 실행이 됩니다. 단 실행이 완료되어도 별도의 UI가 알아서 뜨지는 않고 접속해야 할 로컬 주소가 터미널 창에 표시됩니다. 해당 주소로 접속을 하게 되면, 아래와 같은 UI가 보일 것입니다.

 

- 이제 원하는 자연어나, 이미지 등을 불러와서 이미지를 생성하시면 됩니다. 저는 txt2img 탭에서 단어에서 이미지 생성 기능과, img2img 탭에서 불러온 이미지를 사용하여, 새로운 이미지를 생성하는 두 가지만 사용해보았으나, 그 이외에도 다양한 기능이 있으니, 이는 소스코드를 받은 링크의 설명을 참고하시기 바랍니다.

 

- 위의 이미지는 txt2img 기능만 사용해서 직접 만든 이미지입니다. 키워드들이 전부 반영되기는 하였으나, 실제 원하던 결과물과는 차이가 꽤 있습니다. 그래서 txt2img만으로는 원하는 이미지를 얻기 어렵기 때문에, 반복적으로 생성한 이미지 중에서 원하는 결과물과 비슷한 이미지나 아주 간략하게라도 직접 그린 그림을 가지고  img2img 기능을 사용해서 원하는 비슷한 이미지를 생성하고 이를 다시 소스로 삼아서 반복 실행해야 원하는 이미지를 만들 수 있을 것 같습니다. 

- 마지막으로 본 포스팅을 작성하게 된 유튜브 영상을 첨부합니다. 아래의 영상에서 키워드를 넣는 것과 소스로 어느 정도의 그림을 넣고 어떤 식으로 만들어가는지를 참고하시면 좋을 것 같습니다.

---