본문 바로가기
데이터&AI/LLM

Openai API로 DALL·E 이미지 생성하기!! (feat. python)

by 일등박사 2024. 1. 27.

openai의 api로 gpt3.5 gpt4 를 넘어

langchain, llamaindex 등 다양한 기술을 사용할 수 있는데요!!

 

오늘은 간단하 openai API를 바탕으로 DALL-E 이미지 생성하는 방법을 알아보겠습니다~!

 

DALL·E 이미지 생성 코드 (기본)

코드는 아주 간단합니다!

import openai
client = openai.OpenAI(api_key = '{나의 openai KEY}')
PROMPT = "rabbit wearing swimmingsuit under the sea "


response = client.images.generate(
    prompt=PROMPT,
    n=1,
    size="256x256",
)

 

위의 코드를 실행하면!!

 

이렇게 결과물이 잘 나오구요~~

 

해당 URL을  들어가면! 

(https://oaidalleapiprodscus.blob.core.windows.net/private/org-yyiHvPWrSiGTA1FbCcMmJrQQ/user-sEfZDIexu8Q64OFLoHlNmDcU/img-7WriepCtB9U8DnNwQNRm0Ot1.png?st=2024-01-24T12%3A36%3A10Z&se=2024-01-24T14%3A36%3A10Z&sp=r&sv=2021-08-06&sr=b&rscd=inline&rsct=image/png&skoid=6aaadede-4fb3-4698-a8f6-684d7786b067&sktid=a48cca56-e6da-484e-a814-9c849652bcb3&skt=2024-01-23T22%3A02%3A47Z&ske=2024-01-24T22%3A02%3A47Z&sks=b&skv=2021-08-06&sig=UF2xYzCo4xbksWqOh67/kC202KLqDppf7Oeb48/hhL4%3D)

 

정말 수영복을 입은 토끼가 수영하는 모습의 이미지를 볼 수 있습니다
(아쉽게도 한글은잘 작동하지 않는것 같네요! 

 

 

DALL·E 이미지 생성 코드 (모델 및 크기 설정)

이번엔 모델과 이미지 크기를 default값이 아닌 직접 설정한 값으로 진행해보겠습니다~!

    model="dall-e-3",   size="1024x1024",  quality="standard", 의 설정으로 진행해보지요~!

import openai
client = openai.OpenAI(api_key = '{나의 api key}')
PROMPT = "rabbit wearing swimmingsuit under the sea "


response = client.images.generate(
    prompt=PROMPT,
    model="dall-e-3",
    n=1,
   size="1024x1024",
   quality="standard",
)

역시  결과물은 url로 잘 나왔고~!

https://oaidalleapiprodscus.blob.core.windows.net/private/org-yyiHvPWrSiGTA1FbCcMmJrQQ/user-sEfZDIexu8Q64OFLoHlNmDcU/img-t2PmYfjTget0eWURUv5PWGQh.png?st=2024-01-24T12%3A45%3A27Z&se=2024-01-24T14%3A45%3A27Z&sp=r&sv=2021-08-06&sr=b&rscd=inline&rsct=image/png&skoid=6aaadede-4fb3-4698-a8f6-684d7786b067&sktid=a48cca56-e6da-484e-a814-9c849652bcb3&skt=2024-01-23T22%3A03%3A09Z&ske=2024-01-24T22%3A03%3A09Z&sks=b&skv=2021-08-06&sig=C1/RB9i4FcFkkg5lh3VE9sK63s1MDKAXI761kRPclOU%3D

 

조금 더 고급스러운 고화질의 토끼이미지가 생성됨을 확인할 수 있었습니다~!

 

 

DALL·E 이미지 생성 코드 (여러개 이미지)

한번에 여러개의 이미지를 동시에 만들고 싶다면!!?

n 의 값을 바꾸어주면 됩니다~!

이번에는 3개의 이미지를 만들어봐요!

import openai
client = openai.OpenAI(api_key = '{내 api key}')
PROMPT = "rabbit wearing swimmingsuit under the sea "


response = client.images.generate(
    prompt=PROMPT,
    model="dall-e-2",
    n=3,
 size="256x256",
)

 

결과물에 3개의 이미지정보가있구요~!

 

https://oaidalleapiprodscus.blob.core.windows.net/private/org-yyiHvPWrSiGTA1FbCcMmJrQQ/user-sEfZDIexu8Q64OFLoHlNmDcU/img-rkWyLBou7cnMjjodb8660Hkf.png?st=2024-01-24T12%3A50%3A25Z&se=2024-01-24T14%3A50%3A25Z&sp=r&sv=2021-08-06&sr=b&rscd=inline&rsct=image/png&skoid=6aaadede-4fb3-4698-a8f6-684d7786b067&sktid=a48cca56-e6da-484e-a814-9c849652bcb3&skt=2024-01-23T21%3A59%3A15Z&ske=2024-01-24T21%3A59%3A15Z&sks=b&skv=2021-08-06&sig=pbMNsP49vygkvLJWC83WlLHE7wOpO2gL1tlgaSs6dNc%3D

https://oaidalleapiprodscus.blob.core.windows.net/private/org-yyiHvPWrSiGTA1FbCcMmJrQQ/user-sEfZDIexu8Q64OFLoHlNmDcU/img-mpv9bhZPUOvxbl7aHUiakzgJ.png?st=2024-01-24T12%3A50%3A26Z&se=2024-01-24T14%3A50%3A26Z&sp=r&sv=2021-08-06&sr=b&rscd=inline&rsct=image/png&skoid=6aaadede-4fb3-4698-a8f6-684d7786b067&sktid=a48cca56-e6da-484e-a814-9c849652bcb3&skt=2024-01-23T21%3A59%3A15Z&ske=2024-01-24T21%3A59%3A15Z&sks=b&skv=2021-08-06&sig=XxrnltW1cQVGPx4CS/kI5BCOxxvuoxgPb%2B9XHNncuB8%3D

https://oaidalleapiprodscus.blob.core.windows.net/private/org-yyiHvPWrSiGTA1FbCcMmJrQQ/user-sEfZDIexu8Q64OFLoHlNmDcU/img-mw9bwPgRfE5Ab66BifpFWGvM.png?st=2024-01-24T12%3A50%3A26Z&se=2024-01-24T14%3A50%3A26Z&sp=r&sv=2021-08-06&sr=b&rscd=inline&rsct=image/png&skoid=6aaadede-4fb3-4698-a8f6-684d7786b067&sktid=a48cca56-e6da-484e-a814-9c849652bcb3&skt=2024-01-23T21%3A59%3A15Z&ske=2024-01-24T21%3A59%3A15Z&sks=b&skv=2021-08-06&sig=WfBcYpYFQDwuD6K4stAs1s2uBWsMIXBzp5%2BWY2LvM1A%3D

 

세 주소 모두 귀여운 이미지가 생성되었습니다~!^^

 


이미지 생성의 비용은 아래와 같으니 참고하여 예쁜 이미 만들어보세요!^^

 

ㅁ 참고 : https://platform.openai.com/docs/api-reference/images

 

 

 

 

댓글