基于ChatGPT的图像生成

Chat GPT怎么会忽然大火呢？当我们打开openai的官网时，我们会发现，这个产品已经是第三代迭代的产品，已经默默的开发了三年之久了，很佩服国外大佬的科研和钻研能力。扯远了，今天我们不回顾Chat GPT的历史，也不去深挖Chat GPT的底层技术。我们只看一个可能彻底颠覆普通美工和图片设计工程师们饭碗的一个功能，就是根据描述自动生成图片。
Openai官网已经开放了api的调用，同时提供基于python和node.js的sdk，可以很轻易的调用openai的接口，来生成想要的图片。

首先需要安装openai的lib。

pip install openai

之后直接粘贴下面代码到一个.py文件，文件名随意。

# This is a sample Python script.
import openai
# Press Shift+F10 to execute it or replace it with your code.
# Press Double Shift to search everywhere for classes, files, tool windows, actions, and settings.
import urllib.request

def download_img(img_url):
    request = urllib.request.Request(img_url)
    try:
        response = urllib.request.urlopen(request)
        img_name = "img.png"
        if (response.getcode() == 200):
            with open(img_name, "wb") as f:
                f.write(response.read()) # 将内容写入图片
            return img_name
    except:
        return "failed"

def print_hi():
    openai.api_key = 'your API keys'
    response = openai.Image.create(
        prompt="一只可爱斑点狗",
        n=1,
        size="512x512"
    )
    image_url = response['data'][0]['url']
    download_img(image_url)


# Press the green button in the gutter to run the script.
if __name__ == '__main__':
    print_hi()

# See PyCharm help at https://www.jetbrains.com/help/pycharm/

申请api keys，你可以通过下图所示的地方来申请（点击右上角Personal按钮，然后点击view API keys）

image.png

在获取到api keys之后，便可以运行获得结果。

image.png

看，我们得到了一张可爱的斑点狗的图片，而且，如果我们不满意这张图片，还可以通过修改 n=10,一次性生成10张图片。而且，图片大小也可以选择，当前支持的大小有256x256, 512x512, or 1024x1024像素。

如果我们想要更复杂的图片，可以修改prompt的描述，让他产生更复杂的图片，比如小桥流水人家，就可以生成如下图片：

image.png

得到的图片符合我们的想法。