关于ide:这个开源-AI-项目可以一键去纹身AI我不要面子了么

40次阅读

共计 1413 个字符,预计需要花费 4 分钟才能阅读完成。

转自:机器之心

有些时候,咱们须要把一些人身上的纹身笼罩掉,以防止引人效仿。有的时候人们只是单纯地好奇,想晓得一些大明星如果没有纹身会是什么样子。来自印度的机器学习研究者 Vijish Madhavan 最近开源的一个机器学习工具 SkinDeep 满足了咱们的需要。

网友们也应用这一工具去解决了一些画了重度纹身的人物照片,成果还不错。

该项目标作者 Vijish Madhavan 在看完加拿大歌手贾斯汀 · 比伯的 MV《Anyone》后,萌发了做这个我的项目的打算。贾斯汀 · 比伯在化妆师的帮忙下花了好几个小时的工夫才把他的一身纹身笼罩掉。

MV 视频的成果十分完满,因为制作视频输入是十分艰难的,因而我的项目作者抉择图像来解决。该项目标终点是深度学习是否胜任这项工作,与 photoshop 相比又如何?

我的项目地址:

https://github.com/vijishmadh…

有人会问,为什么不把纹身间接 PS 掉?Photoshop 能够产生十分好的成果,但问题是应用 Photoshop 须要专业知识,如果用 PS 解决纹身的话,你可能须要破费几个小时的工夫去润饰整个图像。

咱们先来看一下成果如何?美国篮球运动员阿伦 · 艾弗森(Allen Iverson)的纹身就是用这个模型去掉的。

下图中第一行为输出图像,第二行为输入图像,输入后果显著感觉到纹身被去除了。

脸部有大量密集纹身的图像,还有其余装璜,AI 的纹身去除成果也十分好:

与业余图像处理软件 photoshop 相比,成果也不错:

看起来 SkinDeep 的成果还不错,但如果纹身是黑白的,还会有一些残留的痕迹。

我的项目介绍

依据作者介绍,实现这个我的项目须要大量的图像对,因为没有适合的数据集,很多时候训练内容采纳合成数据来实现,具体来说:

  • 首先将 APDrawing 数据集图像对与一些背景去掉纹身设计的图像叠加在一起,应用 Python OpenCV 实现;
  • 绘制数据集有线条艺术对,能够模仿纹身线条,这将有助于模型学习和删除这些线条;
  • APDrawing 数据集只有头像,对于全身图像,我的项目作者采纳了以前的我的项目 ArtLine,并将输入与输出图像叠加在一起;

  • ImageDraw.Draw 与森林绿色(forest green colour)色码一起应用,并随机搁置在身材图像上,相似于 fast.ai 中的 Crappify;
  • Photoshop 也被用来在须要蜿蜒和角度扭转的对象上搁置纹身。

这一我的项目是由 Fast.AI 库构建的,你须要装置 fastai 1.0.61 版(及其依赖库),以及 PyTorch 1.6.0,不反对更高的版本。

尝试这一项目标最快办法就是在 Colab 上:

  • https://colab.research.google…

它的输入限度为 500 像素。

限度

去纹身的机器学习模型尽管看起来并不简单,但在事实世界千奇百怪的状况下,有时依然会呈现一些「贴图谬误」的状况。该项目标构建者示意,因为不足数据集反对,所以用于训练的数据集容量无限。另外,如果有人纹了黑白纹身,恐怕人工智能目前还是认不出来的。

如果这个成果被做成网站,或者成为美颜 app 的一个滤镜,那就太好了。最初,SkinDeep 能不能反过来给人加纹身呢?「试穿」的成果或者会火起来。

开源前哨 日常分享热门、乏味和实用的开源我的项目。参加保护 10 万 + Star 的开源技术资源库,包含:Python、Java、C/C++、Go、JS、CSS、Node.js、PHP、.NET 等。

正文完
 0