OpenAI를 이용하여 Image Generator 생성하기

OpenAI의 Dall-E 라는 engine을 이용하여 Image를 자동으로 생성해주는 코드를 작성해보자.

우선 OpenAI의 API가 필요한데 아래 링크를 가면 OpenAI API생성과 python 환경에서 OpenAI API를 연결하는 방법에 대한 기본적인 내용이 들어있다. 

해당 내용을 모른다면, 먼저 아래 포스팅을 읽고오자.

OpenAI/ChatGPT API를 Python에 연동하는 방법 (tistory.com)

OpenAI/ChatGPT API를 Python에 연동하는 방법

OpenAI에는 ChatGPT에서 사용하는 것과 같은 Language model engine도 있지만 Image를 생성해주는 model engine도 존재한다. 

우리는 이 모델을 이용해서 image 생성을 할 것이다.

image 생성 코드는 아래와 같다. 

def generate_code_image(code):
    model_engine = "dall-E"
    prompt = (f"```\n{code}\n```")

    completions = openai.Completion.create(
        engine=model_engine,
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.5,
    )

    message = completions.choices[0].text
    return message

enging, prompt, max_tokens와 같은 파라미터에 대한 자세한 설명은 아래 포스팅을 참고바란다.

OpenAI/ChatGPT를 이용하여 블로그 포스팅 자동화하기 - Pandas 활용 (tistory.com)

OpenAI/ChatGPT를 이용하여 블로그 포스팅 자동화하기 - Pandas 활용

어쨌든 결과적으로 Dall-E engine 사용 시에도 어떤 이미지를 생성하고 싶은지 prompt로 입력을 해주면 된다.

Sample code로 아래와 같이 입력해보았다.

code = """
the earth in 2050
"""

generated_image = generate_code_image(code)

with open("code_image.png", "wb") as f:
    f.write(generated_image)

더 세밀한 명령어를 넣을 수록 좀 더 정교한 이미지를 생성할 수 있다.