跳到主要内容

使用风格迁移快速实现图像滤镜功能

图像滤镜一直是许多应用中常见的功能。使用大模型，可以快速实现图像滤镜功能，不需要复杂的图像处理，就能够快速实现多种滤镜效果。

大模型实现滤镜可以通过 图生图、使用lora、微调模型等多种方式实现。 本期我们使用风格迁移接口快速实现图像滤镜功能。相比其他方式，风格迁移接口使用简单，效果显著，能够快速实现多种滤镜效果。

步骤一：体验接口效果​

功能说明​

Serverless API 提供了接口快速体验的界面，可以快速体验接口效果。 打开 Serverless API 页面 ，找到图像生成与处理下的Kolors模型，点击进入接口详情页，切换到风格迁移的API。

image-20250113153212170

风格迁移接口中，原图指的是想要进行迁移和修改的图片。提示词和image是用以指导原图修改的文本和风格图。

类吉卜力风格迁移​

接下来我们用这个接口 完成 类吉卜力画风 的 风格迁移。

使用下图，作为参考的image。

20250113164303_吉卜力风格__小清新

使用一张普通的风景图作为原图。

P1012997

在页面上填入这些图片，为了让产出的图片更贴合原图，我们讲strength参数设置为0.4。

image-20250113173048253

输出结果如下：

image-20250113173107828

毛茸茸风格迁移​

参考图如下：

image-20250114143124824

和前文步骤相同，我们能够通过风格迁移接口快速复制毛茸茸的图片风格。

输入如下：

image-20250114144653414

输出结果如下：

image-20250114144718497

步骤二：调用接口​

完成了功能体验和验证后，接下来我们可以通过代码调用，将上述功能集成到自己的应用中。

import requests

API_URL = "https://ai.gitee.com/api/serverless/Kolors/image-to-image"
headers = {
    "Authorization": "Bearer XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
    "Content-Type": "application/json"
}

def query(payload):
	response = requests.post(API_URL, headers=headers, json=payload)
	return response.content

output = query({
    "parameters": {
        "prompt": "图片提示词",
        "width": 1024,
        "height": 1024,
        "steps": 25,
        "guidance_scale": 6,
        "strength": 0.5,
        "scale": 0.5
    },
    "image": "参考图的base64字符串",
    "inputs": "原图的base64字符串"
})

with open("output.png", "wb") as file:
	file.write(output)