python调用DALL·E绘画

实现用gpt的api和他对话后,我们试着调用DALL·E的api进行绘画

参考文档

OpenAI API

运行代码

python 复制代码
from openai import OpenAI

client = OpenAI()

user_prompt = input("请输入您想生成的图片描述: ")

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

image_url = response.data[0].url
print(f"生成的图片链接: {image_url}")

运行结果

生成图片

花费了我4美分,生成了以下图片,看起来还算可以

新需求1

接下来我们想要用dall-e-2来对图片进行修改,而且想让python直接显示出图片来,而不是去点击URL

我们需要先下载requestsPillow库,用requests获取图片然后用Pillow来显示,在CMD的当中使用下面的命令去下载即可

python 复制代码
pip install requests Pillow

我们还需要做一张想要改变哪个位置的mask图片,我是用美图秀秀随便扣的,注意要和原图保持相同的尺寸

运行代码

python 复制代码
from openai import OpenAI
import requests
from PIL import Image
from io import BytesIO

client = OpenAI()

user_prompt = input("请问你想如何修改图片: ")

response = client.images.edit(
  model="dall-e-2",
  image=open("E:/Game_Program/勇者.png", "rb"),
  mask=open("E:/Game_Program/mask.png", "rb"),
  prompt=user_prompt,
  size="1024x1024",
  n=1,
)

image_url = response.data[0].url
print(f"生成的图片链接: {image_url}")

# 获取并显示图片
response = requests.get(image_url)
img = Image.open(BytesIO(response.content))
img.show()

运行结果

用了好几美分试了几次,发现生成的图片和第一张还是一样的,不知道哪里出问题了,有兴趣的朋友可以试试

新需求2

接下来我们直接通过上传的图片来生成2张新的图片,这个比较简单,应该不会出错,依旧是用到dall-e-2

运行代码

python 复制代码
from openai import OpenAI
import requests
from PIL import Image
from io import BytesIO

client = OpenAI()

response = client.images.create_variation(
  model="dall-e-2",
  image=open("E:/Game_Program/勇者.png", "rb"),
  size="1024x1024",
  n=2,
)

for data in response.data:
    image_url = data.url
    print(f"生成的图片链接: {image_url}")
    
    # 获取并显示图片
    response = requests.get(image_url)
    img = Image.open(BytesIO(response.content))
    img.show()

运行结果

生成图片

干掉我8美分,生成的两张图片有点一言难尽啊,太抽象了吧

相关推荐
Element_南笙14 分钟前
吴恩达新课程:Agentic AI(笔记2)
数据库·人工智能·笔记·python·深度学习·ui·自然语言处理
倔强青铜三23 分钟前
苦练Python第69天:subprocess模块从入门到上瘾,手把手教你驯服系统命令!
人工智能·python·面试
倔强青铜三28 分钟前
苦练 Python 第 68 天:并发狂飙!concurrent 模块让你 CPU 原地起飞
人工智能·python·面试
星期天要睡觉1 小时前
深度学习——循环神经网络(RNN)实战项目:基于PyTorch的文本情感分析
人工智能·python·rnn·深度学习·神经网络
ERROR_LESS1 小时前
【ADS-1】【python基础-2】基本语法与数据结构(列表、字典、集合)
python
2401_841495641 小时前
【数据结构】基于Floyd算法的最短路径求解
java·数据结构·c++·python·算法··floyd
一晌小贪欢1 小时前
Python爬虫第6课:Selenium自动化浏览器与动态内容抓取
爬虫·python·selenium·网络爬虫·python基础·python3·pathon爬虫
番石榴AI2 小时前
自己动手做一款ChatExcel数据分析系统,智能分析 Excel 数据
人工智能·python·数据挖掘·excel
星期天要睡觉2 小时前
深度学习——循环神经网络(RNN)
人工智能·python·rnn·深度学习·神经网络
Blossom.1183 小时前
把AI“撒”进农田:基于极值量化与状态机的1KB边缘灌溉决策树
人工智能·python·深度学习·算法·目标检测·决策树·机器学习