dalle2 in python

OpenAi에는 이제 공식 API가 있습니다

설치

pip install dalle2

용법

설정

1. https://openai.com/dall-e-2/로 이동하십시오.
2. OpenAI 계정을 만듭니다
3. https://labs.openai.com/으로 이동하십시오.
4. 개발자 도구에서 네트워크 탭을 엽니 다
5. 프롬프트를 입력하고 "생성"을 누릅니다.
6. https://labs.openai.com/api/labs/tasks로 가져 오십시오
7. 요청 헤더에서 승인을 찾은 다음 Bearer 토큰을 얻습니다.

 from dalle2 import Dalle2
dalle = Dalle2 ( "sess-xxxxxxxxxxxxxxxxxxxxxxxxxxxx" ) # your bearer key 

이미지를 생성합니다

 generations = dalle . generate ( "portal to another dimension, digital art" )
print ( generations )

 ✔️ Task created with ID: task-xsuhOthvBXLEjddn3ynyiiOR
⌛ Waiting for task to finish...
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
? Task completed!

[
  {
    'id': 'generation-sCnERSYDPP0Zu14fsdXEcKmL',
    'object': 'generation',
    'created': 1553332711,
    'generation_type': 'ImageGeneration',
    'generation': {
      'image_path': 'https://openailabsprodscus.blob.core.windows.net/private/user-hadpVzldsfs28CwvEZYMUT/generations/generation...'
    },
    'task_id': 'task-nERkiKsdjVCSZ50yD69qewID',
    'prompt_id': 'prompt-2CtaLQsgUbJHHDoJQy9Lul3T',
    'is_public': false
  },
  # 3 more ... 
]

이미지를 생성하고 다운로드하십시오

 file_paths = dalle . generate_and_download ( "portal to another dimension, digital art" )

 ✔️ Task created with ID: task-xsuhOthvBXLEjddn3ynyiiOR
⌛ Waiting for task to finish...
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
? Task completed!
✔️ Downloaded: C:...generation-XySidj4N8EN6Ok9ed15BZ2bs.webp
✔️ Downloaded: C:...generation-IK3UdxDz77FA5SLKpQPIITdU.webp
✔️ Downloaded: C:...generation-uNejKBXz1z6EQxJAT9pAZbof.webp
✔️ Downloaded: C:...generation-Ol1wEqNprf34vNohmJz0iUiE.webp

특정 수의 이미지를 생성합니다

 generations = dalle . generate_amount ( "portal to another dimension" , 8 ) # Every generation has batch size 4 -> amount % 4 == 0 works best

# download images
file_paths = dalle . download ( generations )
print ( file_paths )

 ✔️ Task created with ID: task-lm0V4nZasgAFasd7AsStE67
⌛ Waiting for task to finish...
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
? Task completed!
✔️ Task created with ID: task-WcetZOHt8asdvHb433gi
⌛ Waiting for task to finish...
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
? Task completed!

마스크 파일에서 이미지를 생성합니다

dall · e는 이미지의 투명한 부분을 채우는 "인파 팅"API를 지원합니다. 이 웹 사이트는 기존 이미지를 페인트하는 도구를 제공하여 투명한 부분을 나타내는 부분을 나타냅니다. 이 파이썬 패키지 호출은 제공 한 이미지가 이미 투명 부품을 갖도록 처리되었다고 가정합니다.

 # make the right half of a saved image transparent
from PIL import Image , ImageDraw

image = Image . open ( 'my_image.png' )
m , n = image . size

area_to_keep = ( 0 , 0 , m // 2 , n )
image_alpha = Image . new ( "L" , image . size , 0 )
draw = ImageDraw . Draw ( image_alpha )
draw . rectangle ( area_to_keep , fill = 255 )

image_rgba = image . copy ()
image_rgba . putalpha ( image_alpha )
image_rgba = image_rgba . resize (( 1024 , 1024 )) # image must be square and 1024x1024
image_rgba . save ( 'image_with_transparent_right_half.png' )

# ask DALL·E to fill-in the transparent right half
generations = dalle . generate_from_masked_image (
    "portal to another dimension, digital art" ,
    "image_with_transparent_right_half.png" ,
)

 ✔️ Task created with ID: task-xsuhOthvBXLEjddn3ynyiiOR
⌛ Waiting for task to finish...
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
...task not completed yet
? Task completed!

다른 언어

nodejs 패키지