Python头像动漫化，快来生成一个自己的动漫头像吧

会AIGC的小孩

于 2024-04-09 10:38:06 发布

阅读量544

点赞数 3

文章标签： python 开发语言爬虫学习自动化

本文链接：https://blog.csdn.net/mate1357/article/details/137542088

版权

本文介绍如何使用Python和百度AI的API，通过获取access_token和调用人像动漫化接口，实现用户头像的动漫化转换，包括获取token的步骤和实际代码示例。

摘要由CSDN通过智能技术生成

一、前言

很多时候我们都会为头像发愁，像我这种万年不换头像的咸鱼从来没有这种烦恼。但是吧，有个个性化的头像还是非常有趣的，因为这个抠鼻屎的头像除了抠鼻屎这点，其它都很符合本人的气质，所以已经够用了。但是我还是打算把这个小技巧给大家分享。

二、效果展示

在我多次测试后，发现女生头像的生成还是比较不错的，而且最好是正面照。而男生动漫头像的生成就有点差强人意了，大家可以自己试试，下面我们看看我用有村大妹子生成的两张图片，我们先看第一张：在这里插入图片描述接下来是第二张，我用的是垫底辣妹中的形象：两张效果都还可以，主要还是人物的头部动漫化了，其它部分只是趋于动漫的风格。不过我觉得还是很可以的。

三、实现

对于我这种基本功不扎实的人来说，实现这种只能调用API了。我使用的是百度的API，我们可以进入百度AI开放平台注册账号，具体流程就不讲了。登陆后我们可以看到如下界面：在这里插入图片描述我们依次点击开放能力》图像技术》人像动漫化，然后我们点击立即使用，再创建应用就可以了。在应用里面我们可以看到三个参数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')

这样我们就实现了头像动漫化的操作，非常简单，因为主要的工作都由百度的程序员帮我们实现了，大家可以使用自己的照片生成一些个性化的头像。如果太丑的话不要跑回来骂我就好了。