水墨江南模型Python入门实战：10行代码生成第一幅水墨画

想用AI画一幅充满诗意的水墨画，是不是觉得门槛很高，需要复杂的代码和配置？今天，我们就来打破这个刻板印象。无论你是刚接触Python的新手，还是对AI绘画感兴趣的爱好者，这篇文章都将带你用最简单、最直接的方式，快速上手“水墨江南”风格图像生成。整个过程，核心代码只需要10行左右，你就能亲眼看到AI如何将一段文字描述，变成一幅韵味十足的水墨画。

我们的目标很明确：不涉及复杂的理论，不搞繁琐的环境搭建，就是手把手地带你跑通第一个生成案例。你会学到如何安装必要的工具，如何调用模型，以及如何调整一句话，让生成的画面更符合你的想象。准备好了吗？让我们开始这段充满古典美学的AI创作之旅。

1. 环境准备：安装你的“画笔”和“颜料”

在开始画画之前，我们得先准备好画具。对于用Python调用AI模型来说，画具就是几个必要的软件库。别担心，安装过程就像在手机上下载App一样简单。

首先，你需要确保电脑上已经安装了Python。建议使用Python 3.8或更高的版本。打开你的命令行工具（Windows上是CMD或PowerShell，Mac或Linux上是终端），输入 python --version 或 python3 --version 来检查。如果看到了版本号，比如“Python 3.10.0”，那就没问题了。

接下来，我们安装两个核心的“画笔”库。第一个是 requests，它负责帮我们的程序去和远端的AI模型“对话”，发送请求和接收图片。第二个是 Pillow（PIL），它是一个非常强大的图像处理库，我们用它来打开和保存AI生成的画作。

在命令行里，输入下面这行命令，然后按回车：

pip install requests Pillow

如果系统提示权限不足，可以尝试在前面加上 sudo（Mac/Linux）或者使用 pip install --user 选项。稍等片刻，看到“Successfully installed”的字样，就说明安装成功了。现在，你的数字画具箱就已经准备就绪了。

2. 核心概念：一句话理解AI绘画

在写代码之前，我们花一分钟了解一下AI绘画是怎么工作的。你可以把它想象成一位技艺高超但需要明确指令的画师。

这位“AI画师”的核心能力，是理解你用自然语言写的“绘画要求”，然后根据它学过的海量水墨画知识，创作出一幅全新的作品。这个“绘画要求”，在技术领域被称为“提示词”（Prompt）。提示词写得好不好，直接决定了最终画作的质量和是否符合预期。

比如，如果你只说“画一座山”，AI可能会生成任何风格的山。但如果你说“用宋代水墨画风格，画一座云雾缭绕的远山，近处有松树，留白要多”，AI画师就能更精准地捕捉到你想要的那种古典、空灵的意境。

我们今天要调用的“水墨江南”模型，就是一位专门训练来绘制中国水墨风格画的AI画师。它特别擅长理解那些富有诗意的中文场景描述。理解了这个核心——用文字描述驱动图像生成，后面的代码就会变得非常直观。

3. 10行代码实战：生成你的第一幅水墨画

现在，进入最激动人心的环节：写代码。我们将整个过程分解为清晰的四步，你可以跟着一步一步来。

3.1 第一步：导入工具库

打开你喜欢的代码编辑器（比如VS Code、PyCharm，甚至系统的记事本都可以），新建一个Python文件，命名为 create_ink_painting.py。然后在文件的最开始，写下这两行代码：

import requests
from PIL import Image
import io

• import requests: 引入我们之前安装的“通信员”，负责和AI模型服务器联系。
• from PIL import Image: 从Pillow库中引入处理图像的核心功能。
• import io: 这是一个Python自带的库，用于在内存中处理二进制数据（比如图片数据），不需要额外安装。

3.2 第二步：准备绘画指令和访问密钥

接下来，我们需要告诉AI两件事：画什么，以及我们是谁（用于身份验证）。这里假设你已经从某个AI模型服务平台（例如百度智能云、阿里云、或专门的AI绘画平台）获得了API访问密钥（API Key）和模型的调用地址（Endpoint）。

# 你的API密钥和请求地址（此处需要替换为你自己的）
API_KEY = "你的实际API密钥"
API_URL = "https://api.example.com/v1/images/generations" # 示例地址，需替换

# 你想要生成的画面描述
prompt = "江南水乡，白墙黛瓦，小桥流水，一艘乌篷船，细雨朦胧，水墨风格"

重要提示：

• 你需要将 API_KEY 和 API_URL 替换成从你所使用的平台获取的真实信息。不同的平台，获取这些信息的方式和地址格式会有所不同，请查阅对应平台的文档。
• prompt 变量里的文字，就是你的“绘画指令”。你可以自由修改它，比如改成“孤舟蓑笠翁，独钓寒江雪”或者“黄山奇松，云海翻腾”，来生成不同主题的画作。

3.3 第三步：调用AI模型并获取结果

现在，我们让“通信员”（requests）带着你的指令和身份证明，去拜访“AI画师”。

# 设置请求头，携带认证信息
headers = {
    "Authorization": f"Bearer {API_KEY}",
    "Content-Type": "application/json"
}

# 构建请求数据，告诉AI模型我们要什么
data = {
    "prompt": prompt,
    "n": 1,  # 生成1张图片
    "size": "1024x1024"  # 图片尺寸
}

# 发送POST请求
response = requests.post(API_URL, headers=headers, json=data)

这段代码做了以下几件事：

1. 在 headers 里，用 Bearer {API_KEY} 的格式声明了我们的合法身份。
2. 在 data 里，打包了我们的具体需求：绘画指令（prompt）、生成图片数量（n=1）、图片尺寸（size）。
3. 最后一行，使用 requests.post 方法，将所有这些信息发送到指定的API地址。

3.4 第四步：处理并保存生成的画作

AI画师收到请求后，会开始创作，并把画好的作品（一张图片）寄回来。我们需要接收并妥善保存它。

# 检查请求是否成功
if response.status_code == 200:
    # 解析返回的JSON数据，获取图片的URL
    result = response.json()
    image_url = result['data'][0]['url']  # 根据实际API返回结构调整

    # 从URL下载图片数据
    img_response = requests.get(image_url)
    img_data = img_response.content

    # 使用PIL打开图片并保存到本地
    image = Image.open(io.BytesIO(img_data))
    image.save("我的第一幅水墨画.png")
    print("恭喜！水墨画已生成，保存为‘我的第一幅水墨画.png’")
else:
    print("请求失败，状态码:", response.status_code)
    print("错误信息:", response.text)

• if response.status_code == 200:：这是一个状态码，200表示服务器成功处理了我们的请求。
• image_url = result['data'][0]['url']：从返回的复杂数据中，找到生成图片的下载链接。请注意，不同的AI平台API返回的数据结构可能不同，这里的 ['data'][0]['url'] 是常见格式之一，你需要根据自己所用平台的官方文档进行调整。有时返回的可能是直接的图片二进制数据，而非URL。
• Image.open(io.BytesIO(img_data))：将从网络下载的图片二进制数据，转换成PIL可以处理的图像对象。
• image.save(...)：最后，将这幅珍贵的作品保存到你电脑的当前目录下。

将以上所有代码段按顺序组合在一起，就是完整的脚本。现在，在命令行中，进入到你的代码文件所在目录，运行：

python create_ink_painting.py

如果一切顺利，你会在当前文件夹里看到一个名为“我的第一幅水墨画.png”的新文件，打开它，你的第一幅AI水墨画就诞生了！

4. 让画作更精彩：提示词微调技巧

成功运行第一次后，你可能会想：我能控制画面内容吗？答案是肯定的。关键在于“提示词”的编写。这里分享几个简单易懂的技巧：

1. 主体+细节+风格：这是最基础的公式。例如，“主体：一只小猫；细节：蹲在青瓦屋檐下，尾巴卷曲；风格：水墨画，淡雅风格”。这比单纯的“一只猫”要生动得多。
2. 善用风格词：“水墨江南”模型本身已限定风格，但你还可以叠加更细致的描述，如“写意水墨”、“工笔水墨”、“宋代山水画风格”、“带有书法飞白笔触”。
3. 控制画面情绪：通过形容词和场景设定来传达氛围。比如，“宁静的月下竹林”和“狂风暴雨中的竹林”，生成的画面意境会截然不同。
4. 尝试否定词：有些平台支持“负面提示词”，告诉AI你不想要什么。例如，在提示词后补充“不要现代建筑，不要鲜艳色彩”。

你可以多复制几份代码文件，每次只修改 prompt 变量里的文字，然后分别运行，观察不同描述带来的画面变化。这是探索AI绘画乐趣最快的方式。

5. 常见问题与小贴士

第一次尝试，可能会遇到一些小麻烦。这里列出几个常见问题及解决方法：

• 问题：运行代码后报错，提示ModuleNotFoundError: No module named 'requests'。 解决：这说明requests库没有安装成功。请回到命令行，确认是否在正确的Python环境下执行了 pip install requests Pillow。

• 问题：请求失败，状态码是401或403。 解决：这几乎总是因为API_KEY或API_URL不正确，或者API密钥已过期、没有权限。请仔细检查并替换成从你的AI服务平台获取的有效信息。

• 问题：代码运行成功，也生成了文件，但图片打不开或是一片空白/黑色。 解决：首先检查图片大小是否正常。如果大小只有几KB，可能意味着API调用实际失败了，但程序没有正确判断。请打印出 response.text 看看服务器返回的具体错误信息。另外，请确保你解析图片URL或数据的方式，符合该平台API文档的说明。

• 小贴士：在正式生成大量图片前，可以先将图片尺寸size设置为较小的如“512x512”，这样生成速度更快，用于测试提示词效果也更经济。

---

获取更多AI镜像

想探索更多AI镜像和应用场景？访问 CSDN星图镜像广场，提供丰富的预置镜像，覆盖大模型推理、图像生成、视频生成、模型微调等多个领域，支持一键部署。