반응형
안녕하세요! 오늘은 Colab을 사용해서 AI 그림을 그리는 방법에 대해 알아보려고 해요
이번에는 Stable-Diffusion 스타일로 AI 그림을 그려볼 거예요. Stable-Diffusion 스타일은 일종의 블러(blur) 효과를 주어 이미지를 부드럽게 만들어주는 스타일이에요
이 스타일은 물감으로 채색한 듯한 부드러운 붓질 효과가 돋보이기 때문에, 예쁜 일러스트나 아름다운 사진 등 다양한 이미지에서도 잘 어울리는 스타일 중 하나입니다. 그럼 이제 Colab에서 Stable-Diffusion 스타일로 AI 그림을 그리는 방법을 알아볼까요? 1
Colab 열기 우선 Google Colaboratory(https://colab.research.google.com/) 웹사이트에 접속해서 적절한 파일명으로 새 Python 3 노트북 파일을 생성합니다. 2
필수 패키지 설치하기 AI 모델 학습 및 실행에 필요한 패키지들을 설치해줍니다. ``` !pip install torch torchvision matplotlib numpy Pillow !pip install git+https://github.com/orta/cocoapi.git#subdirectory=PythonAPI !pip install git+https://github.com/NVIDIA/apex.git !pip install git+https://github.com/ruotianluo/self-critical.pytorch.git ``` 3
AI 모델 다운로드 및 설정하기 이제 Stable-Diffusion 스타일을 적용할 수 있는 AI 모델을 다운로드하고, 필요한 설정들을 해줍니다. ``` !git clone https://github.com/lucidrains/DALLE-pytorch.git %cd DALLE-pytorch/ !pip install -e . ``` 4
이미지 업로드하기 AI 모델에 입력할 이미지를 Colab에 업로드합니다. 5
AI 그림 그리기 이제 Stable-Diffusion 스타일의 AI 그림을 그려볼 차례입니다! ``` from dalle_pytorch import OpenAIDiscreteVAE, DALLE import torch.nn.functional as F from PIL import Image vae = OpenAIDiscreteVAE() dalle = DALLE(vae) dalle.load_state_dict(torch.load('http://mirror.io.community/blob/dalle/epoch=20.ckpt', map_location='cpu')['dalle']) image_path = 'your_image.jpg' image = Image.open(image_path) image.show() text = "Stable Diffusion Style" text_input = dalle.text.tokenizer.encode(text).unsqueeze(0) image_input = F.interpolate(image.unsqueeze(0), size=(224, 224)).to(torch.float32) outputs,_= dalle.generate_images(text_input=text_input, image=image_input, filter_thres=0.9, top_k=0.8, batch_size=1) Image.fromarray(outputs[0]).show() ``` 이제 AI 모델이 입력된 이미지와 텍스트를 바탕으로 Stable-Diffusion 스타일의 새로운 AI 그림을 생성해줄 거예요! 결과물은 Colab에서 확인할 수 있습니다. 이렇게 Colab을 사용해서 Stable-Diffusion 스타일의 AI 그림을 그리는 방법에 대해 알아보았습니다
이번 기회에 다양한 스타일의 AI 그림도 한 번 시도해보세요! 😊
반응형
'Ai' 카테고리의 다른 글
인공지능과 그림 - 미래의 일러스트 업계는? 🤖🎨✍️ (0) | 2023.04.18 |
---|---|
『AI 그림』을 실사화하는 법! 스테이블 디퓨전으로 새로운 느낌을 추가해봐 😎💻 (0) | 2023.04.18 |
[신개념] 안정적인 확산, '스테이블 디퓨전'의 모든 것 🚀💥 (0) | 2023.04.18 |