Colab Stable Diffusion 그림그리기

 

Untitled0.ipynb 시작단계에서 반드시 런타임을 클릭하여 런타임 유형변경에서 GPU를 설정하자.

 

2023년 2월 7일 기준 tensorflow가 불과 며칠 전에 버전 2.92였던 것이 2.11로 업그레이드 되었다. 하지만 default 버전을 출력해 보면 여전히 2.9.2로 출력된다. 여타의 머신러닝에서는 문제가 되지 않지만 Stable Diffuaion에 의한 그림 그리기작업에서 tensorflow 버전이 맞지 않는다는 에러 메세지가 발생하여 뜬다.

1	import tensorflow as tf print(tf.__version__) ----------------------------------------------------------- 2.9.2

 

아래와 같이 TensorFlow 를 2.15 버전으로 설치하자. 출력 결과를 보면

tensorflow 2.15 와 keras 2.15 가 함께 설치됨을 알 수 있다. 아울러 런타임에서 런타임 다시시작을 클릭하자.

2	pip install tensorflow=2.15 ----------------------------------------------------------- ...

 

가장 중요한 keras-cv를 설치하자. 

3	!pip install keras_cv ----------------------------------------------------------- ... Installing collected packages: keras_cv Successfully installed keras_cv-0.4.2

 

파이선 라이브러리 time, keras_cv, keras, matplotlib.pyplot 라이브러리들을 설치하자.

4	import time import keras_cv from tensorflow import keras import matplotlib.pyplot as plt

 

그림을 그릴수 있도록 그래픽 규격을 설정한다.

5	def plot_images(images): plt.figure(figsize=(20,20)) for i in range(len(images)): ax = plt.subplot(1, len(images), i+1) plt.imshow(images[i]) plt.axis("off") plt.tight_layout()

 

keras_cv,models를 상속하는 class StableDiffusion을 model로 설정하자.

6	model = keras_cv.models.StableDiffusion(img_width=512, img_height=256)

 

그림의 스타일을 지정하는 Prompt를 잘작성하여 넣은 후 코드를 실행하자.

최초 실행인 경우 서버로부터 binary 파일 형태의 사전학습된 내용을 다운받는 시간이 수분 소요되며 그림 한 장당 1~2분 안에 출력이 이루어진다.

7	images = model.text_to_image("photograph of a person walking along the narrow street rainy day in the style of Van Gogh's Saint Remy", batch_size=3) plot_images(images)

 

첫번째와 두번째 그림이 고호풍으로 그려진 그럴듯한  그림임을 인정할 수 밖에 없을 것이다. 두번째 그림의 아저씨는 그 유명한 고호 자화상에서 따온 그림인듯하다.

 

첨부된 파일을 다운받아 Colab에서 Prompt를 조금씩 바꾸어 실행해보자.

StableDiffusion.ipynb

15.89MB