openai PIL 이미지 생성 코딩 방법

openai의 api와 파이썬 이미지 라이브러리인 PIL로 텍스트 입력을 받으면 이미지를 생성해 주는 코드를 만들어 보자.

 

 

openai PIL 이미지 생성

참고로 PIL(Python Image Library)은 파이썬 이미지 라이브러리의 약자이다. 이 PIL 라이브러리를 사용하면 다양한 이미지 파일의 형식을 생성하거나 편집 또는 조작등이 가능하다. 또한 openai의 라이브러리는 아니지만 이 파이썬 라이브러리와 함께 openai의 라이브러리, 예를 들면 최근 베타 버전으로 발행된 DELL-E 2 모델을 조합하여 이미지를 조작도 가능하다.

 

 

하지만 현재 DELL-E 모델은 베타 버전이며 실제 사용하려면 결제를 해야 하는 번거로움이 있다. 따라서 openai 를 import 해서 PIL 라이브러리로 샘플 코드를 사용 중이다.

 

다음 샘플 코드는 하드코딩으로 생성하고자 하는 이미지의 prompt 입력을 받으면 해당 이미지를 생성해 주는 코드로, 흰 샴고양이를 prompt로 받아 이미지를 생성해 주는 코드이다.

 

import openai
import requests
from PIL import Image

# Set up OpenAI API key
openai.api_key = "API-KEY"

# Prompt for image generation
prompt = "a white siamese cat"

# Generate image using DALL-E 2 API
response = openai.Image.create(
    prompt=prompt,
    n=1,
    size="512x512",
    model="image-alpha-001",
    response_format="url"
)

# Get image URL from response
image_url = response["data"][0]["url"]

# Download image from URL
image_data = requests.get(image_url).content

# Save image to file
with open("generated_image.jpg", "wb") as f:
    f.write(image_data)

# Display image
image = Image.open("generated_image.jpg")

image.show()

 

다음 이미지는 prompt = "a white siamese cat"을 입력으로 받아 ai가 생성한 흰 샴 고양이 이미지이다.

 

white-siamese-cat

 

파이썬 관련 다른 글 보기

Pillow 파이썬 이미지 처리 라이브러리