专业的编程技术博客社区

网站首页 > 博客文章 正文

青灯教育|女朋友找你要动漫头像,Python助你实现

baijin 2024-08-10 13:43:37 博客文章 18 ℃ 0 评论

很多时候你是否为头像发愁过,当然像我这样万年不换头像的咸鱼就没有这样的烦恼。但是如果拥有个性化的头像也是非常有趣的一件事,例如使用自己的漫画头像?今天给大家分享如何用Python实现个人头像动漫化。

展示效果

从效果上看,主要是对人物的头像动漫画了,其他的部分。。。不过我觉得还是可以接受的吧。

具体实现

实现这种功能从底层代码上来讲,对于我来说实属超纲了,在这里我们主要讲如何使用第三方API来实现动漫化。这里使用的是百度的API,具体注册百度API开放平台就不说了,在百度AI平台中我们找到需要的功能。

选择人像动漫化功能后,创建应用,从demo示例代码中我们可以看到三个参数AppID、API Key和Secret Key。

实现人像动漫化需要用到两个接口,一个用来获取access_token,另一个则是用来获取动漫头像,我们来获取access_token:

import requests

def get_access_token():

# 获取token的API

url = 'https://aip.baidubce.com/oauth/2.0/token'

# 获取access_token需要的参数

params = {

# 固定参数

'grant_type':'client_credentials',

# 必选参数,传入你的API Key

'client_id':'你的API Key',

# 必选参数,传入你的Secret Key

'client_secret':'你的Secret Key'

}

# 发送请求,获取响应数据

response = requests.post(url, params)

# 将响应的数据转成字典类型,然后取出access_token

access_token = eval(response.text)['access_token']

# 将access_token返回

return access_token

注意这里我们使用了requests模块,如果没有安装的可以执行以下命令安装:

pip install requests

接下来就是我们头像动漫化了:

import base64

import requests

def img2Cartoon(img):

# 头像动漫化的API

url = 'https://aip.baidubce.com/rest/2.0/image-process/v1/selfie_anime'

# 以二进制的方式读取原始图片

origin_im = open(img, 'rb')

# 将图片进行base64编码

img = base64.b64encode(origin_im .read())

# 关闭原图片

origin_im.close()

# 请求的headers信息,固定写法

headers = {'content-type':'application/x-www-form-urlencoded'}

# 请求的参数

params = {

# 开始获取的access_token

'access_token':get_access_token(),

# 图片的base64编码

'image':img,

}

# 发送请求

response = requests.post(url, data=params, headers=headers)

# 对响应结果进行处理

if response:

# 打开一个文件

f = open('result.jpg', 'wb')

# 获取动漫头像

anime = response.json()['image']

# 对返回的头像进行解码

anime = base64.b64decode(anime)

# 将头像写入文件当中

f.write(anime)

f.close()

我们只需要在main中调用img2Cartoon函数即可完成动漫头像的生成,完整代码如下:

import base64

import requests

def get_access_token():

# 获取token的API

url = 'https://aip.baidubce.com/oauth/2.0/token'

# 获取access_token需要的参数

params = {

# 固定参数

'grant_type':'client_credentials',

# 必选参数,传入你的API Key

'client_id':'你的API Key',

# 必选参数,传入你的Secret Key

'client_secret':'你的Secret Key'

}

# 发送请求,获取响应数据

response = requests.post(url, params)

# 将响应的数据转成字典类型,然后取出access_token

access_token = eval(response.text)['access_token']

# 将access_token返回

return access_token

def img2Cartoon(img):

# 头像动漫化的API

url = 'https://aip.baidubce.com/rest/2.0/image-process/v1/selfie_anime'

# 以二进制的方式读取原始图片

origin_im = open(img, 'rb')

# 将图片进行base64编码

img = base64.b64encode(origin_im .read())

# 关闭原图片

origin_im.close()

# 请求的headers信息,固定写法

headers = {'content-type':'application/x-www-form-urlencoded'}

# 请求的参数

params = {

# 开始获取的access_token

'access_token':get_access_token(),

# 图片的base64编码

'image':img,

}

# 发送请求

response = requests.post(url, data=params, headers=headers)

# 对响应结果进行处理

if response:

# 打开一个文件

f = open('result.jpg', 'wb')

# 获取动漫头像

anime = response.json()['image']

# 对返回的头像进行解码

anime = base64.b64decode(anime)

# 将头像写入文件当中

f.write(anime)

f.close()

if __name__ == '__main__':

img2Cartoon('origin.jpg')

这样我们就实现了人像动漫化,过程还是非常简单的,主要就是使用第三方API实现功能,大家可以使用自己的照片做一些个性化的头像。

Tags:

本文暂时没有评论,来添加一个吧(●'◡'●)

欢迎 发表评论:

最近发表
标签列表