当前位置:首页 > 资讯 > 正文

使用百度AI接口自制图像风格转换数据集

使用百度AI接口自制图像风格转换数据集

  前一段时间在做毕业设计,打算做一些风格转换相关的项目,想自己制作一份数据集用于训练,于是发现了百度ai平台提供的api接口可以实现“人脸动漫化”、“图像风格转换”等功能。按照一定步骤操作后,只要通过post请求就可以将本地的图片进行风格转换了。感觉还不错,比直接使用现有数据集好多了,起码数据质量上可以自己保证~

  当然按照教程来做也非常简单,不过还是做个记录,下次就不用研究教程了。

记录一下操作步骤:

  1. 打开:

  里面有很多功能,我使用的图像转换和人脸动漫化(其实都差不多),选择图像风格转换进入。

  接下来可以看到各种提供的图像风格转换的效果,我们直接使用,点开“产品价格”,往下可以看到有免费几百次的使用机会,点击“立即使用”。

  现在来到了“百度智能云”界面,先注册一个账号,并登录,来到主页:

  先“领取免费资源”,进去后选择你想要的尝试的功能,我选了“图像转换”和“人脸动漫化”之类的功能。(反正也都是免费选,大不了全选了)

  然后出来之后“创建应用”(我这里已经有一个了),名称和描述都随意,把需要的功能选上就行(我选了图像转换、人脸动漫化等)。

获得AK和SK

点击“管理应用”,可以看到自己的AK和SK

拿到这两个字符串后,下一步获取token。

获得token

运行下面代码(注意替换成自己的AKSK):

 

注意替换代码里的AK和SK。

运行后会得到一个json串,里面的refresh_token字段就是我们的token。把它复制保存起来,我们马上就要用。

有了token就可以使用post请求进行图像风格转换测试了:

模板:

 

注意替换代码里的token。这里test.png图片可以自己随便选一张照片放在相同目录下,运行之后,就会在相同目录下得到结果res_mononoke.png。

我的效果如下:

 

  感觉这个风格还是蛮帅的,后面风格转换也打算以此作为参照来生成数据集,一般购买一千次调用次数应该就够用于训练了。