当前位置:   article > 正文

openai中的DALL-E2生成图像_java对接 dall-e2

java对接 dall-e2

使用步骤

使用 OpenAI 的 DALL-E 2 生成图像需要以下步骤:

  1. 准备文本描述:需要一个文本描述来描述所需生成的图像。

  1. 发送 API 请求:使用对应语言的 API 库向 OpenAI API 发送请求,并将文本描述作为请求参数。

  1. 解析 API 返回:从 API 返回中解析生成的图像并保存到本地。

代码举例

Java

  1. import java.io.IOException;
  2. import java.io.InputStreamReader;
  3. import java.io.Reader;
  4. import java.net.HttpURLConnection;
  5. import java.net.URL;
  6. import java.util.HashMap;
  7. import java.util.Map;
  8. import com.google.gson.Gson;
  9. public class OpenAIDalle2 {
  10. public static void main(String[] args) throws IOException {
  11. // 发送 API 请求
  12. URL url = new URL("https://api.openai.com/v1/images/generations");
  13. HttpURLConnection con = (HttpURLConnection) url.openConnection();
  14. con.setRequestMethod("POST");
  15. con.setRequestProperty("Content-Type", "application/json");
  16. con.setRequestProperty("Authorization", "Bearer <API_KEY>");
  17. con.setDoOutput(true);
  18. // 请求参数
  19. Map<String, Object> requestJson = new HashMap<>();
  20. requestJson.put("model", "image-alpha-001");
  21. requestJson.put("prompt", "A four-legged animal that barks");
  22. requestJson.put("num_images", 1);
  23. requestJson.put("size", "1024x1024");
  24. // 发送请求
  25. con.getOutputStream().write(new Gson().toJson(requestJson).getBytes());
  26. // 解析 API 返回
  27. Reader reader = new InputStreamReader(con.getInputStream());
  28. OpenAIDalle2Response response = new Gson().fromJson(reader, OpenAIDalle2Response.class);
  29. // 保存图像到本地
  30. // ...
  31. }
  32. private static class OpenAIDalle2Response {
  33. String data;
  34. // ...
  35. }
  36. }

nodejs

安装客户端

使用Node.js生成图像,需要先安装OpenAI API客户端

npm install openai
导入OpenAI API客户端
const openai = require("openai");
配置您的API密钥
openai.promise.setApiKey("<API_KEY>");
生成图像
  1. openai.ImageCompletion.create({
  2. prompt: "A 3-story red brick building with a white door and windows. In front of the building, there is a green tree with a swing hanging from it.",
  3. n: 1,
  4. size: "1024x1024",
  5. response_format: "url"
  6. }).then((image) => {
  7. console.log(image.data[0].url);
  8. }).catch((error) => {
  9. console.error(error);
  10. });

python

安装OpenAI API客户端
pip install openai
导入OpenAI API客户端
import openai
配置您的API密钥
openai.api_key = "<API_KEY>"
生成图像
  1. prompt = "A 3-story red brick building with a white door and windows. In front of the building, there is a green tree with a swing hanging from it."
  2. completion = openai.Completion.create(
  3. engine="image-alpha-001",
  4. prompt=prompt,
  5. max_tokens=1024,
  6. n=1,
  7. stop=None,
  8. temperature=0.5,
  9. )
  10. print(completion.choices[0].text)
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/575196
推荐阅读
相关标签
  

闽ICP备14008679号