【Python】一行python代码利用人工智能去除工作照背景

在做班组一线风采PPT时，发现其中一位美女没有白底照片，这对一个处女座的PPTer是无法忍受的，那么我们如何用人工智能的方式来解决这个问题呢？

此时我们需要用到的python库是backgroundremover。

官方网址是：

https://github.com/nadermx/backgroundremover

官方介绍是：

可以看出backgroundremover不仅可以删除照片背景还可以删除视频背景，由于涉及到人工智能和视频，所以backgroundremover需要pytorch和ffmpeg的支持。

废话不多说，我们直接上代码。

第一步：安装库

pip install backgroundremover

如果安装成功，会出现以下文件夹：C:\Users\Administrator\.u2net

第二步：利用python通过os执行终端命令

import os
os.system('backgroundremover -i "123.jpg" -o "123_new.jpg"')

在第二步时，初次使用需要下载模型（168 MB），下载地址是google，因此需要能够访问google，才能下载。

我已经给你下载好了，链接: 

https://pan.baidu.com/s/1FbUIBJsEua7G1vYo06lSmg 提取码: o0uu

可获取模型文件：u2net.pth

C:\Users\Administrator\.u2net这个文件夹里放入下载好的u2net.pth文件就算离线安装好了预训练模型。

样例效果：

可以看出去除背景后，并非替换成白色背景，而是背景变为透明底，同理可以把其他照片的白底全部去除。

效果明显更酷炫了。

往期精彩回顾




适合初学者入门人工智能的路线及资料下载中国大学慕课《机器学习》（黄海广主讲）机器学习及深度学习笔记等资料打印机器学习在线手册深度学习笔记专辑《统计学习方法》的代码复现专辑
AI基础下载本站qq群955171419，加入微信群请扫码：