[API] DALL·E 2 Python으로 사용해보기

ChatGPT를 개발한 OpenAI는 작년 4월에 Text to Image 모델인 DALL·E 2도 공개한 적이 있습니다.

최근에 그림을 그려주는 AI와 사진을 아니메 (Anime) 풍으로 변환을 해주는 각종 필터가 유행을 하고 있어서 이미지와 관련된 AI는 놀랍게 느껴지진 않을 수 있습니다. 하지만 단순한 설명만으로 저작권 없는 고퀄리티 이미지를 생성할 수 있기 때문에 아트 업계 쪽에서는 큰 위협이 될 수 도 있습니다. 개인적으로 AI가 불러오는 disruptive innovation의 가장 첫 번째 분야는 예술이 아닐까 싶습니다. OpenAI 홈페이지에서 체험을 하시려면 아래의 링크를 참고하시면 됩니다

DALL·E

그럼 이제 Python으로 DALL·E 2를 사용하는 방법에 대해 알아보겠습니다.

---

Python으로 구현

OpenAI에서 발급받은 API Key를 사용하여 아래와 같이 코드를 짜시면 됩니다.

(API Key 발급 방법을 모르시겠다면 이 포스트를 참고해 주세요)

 

import os
import openai
import json
from pathlib import Path
from base64 import b64decode

description = 'a flying hamburger in space, digital art'    # Ai가 그려줄 이미지에 대한 묘사
response = openai.Image.create(prompt=description            # 텍스트 입력
                               , n=1                    # 생성할 이미지 개수
                               , size= "1024x1024"      # 이미지 크기
                               , response_format= "b64_json"
                               , api_key = 'API KEY 16자리 입력') # openai API키

 
DATA_DIR = Path.cwd() # os.getcwd()
file_name = DATA_DIR / f"{PROMPT[:5]}_{response['created']}.json"  # 생성된 이미지파일 저장 경로

# json 파일로 저장하기 
with open(file_name, mode="w", encoding="utf-8") as file:
    json.dump(response, file)
    
# png 파일로 저장
for index, image_dict in enumerate(response["data"]):
    image_data = b64decode(image_dict["b64_json"])
    image_file = DATA_DIR / f"{file_name.stem}-{index}.png"
    with open(image_file, mode="wb") as png:
        png.write(image_data)

 

 

결과물

위의 코드를 실행하게 되면 아래와 같은 이미지와 json 파일이 생성됩니다.

 

a flying hamburger in space

 

뭔가 사실적이고 그럴듯합니다. 하지만 아직 사람이나 동물의 사실적인 묘사는 퀄리티가 조금 아쉬운 부분도 있습니다.

아래에는 제가 생성한 이미지의 모음입니다. 중간중간 이상하고 어색한 부분도 있어서 아직까지는 불쾌한 골짜기에 놓인 분야도 있는 것 같습니다. AI가 세상을 지배하는 건 먼 훗날의 이야기일 것 같습니다 ㅎㅎ

 

생성 예시 모음

- high-teen north korean leader kim jeong eun, anime style

- a dog and a cat fighting with light saber, digital art

- teddy bears shopping for groceries, one-line drawing

 

- a man riding a tiger, pixel art

- a spacious futuristic library filled with starfishes reading books

- a house in the shape of a pineapple