社区所有版块导航
Python
python开源   Django   Python   DjangoApp   pycharm  
DATA
docker   Elasticsearch  
aigc
aigc   chatgpt  
WEB开发
linux   MongoDB   Redis   DATABASE   NGINX   其他Web框架   web工具   zookeeper   tornado   NoSql   Bootstrap   js   peewee   Git   bottle   IE   MQ   Jquery  
机器学习
机器学习算法  
Python88.com
反馈   公告   社区推广  
产品
短视频  
印度
印度  
Py学习  »  Python

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

机器学习初学者 • 4 年前 • 514 次点击  
在做班组一线风采PPT时,发现其中一位美女没有白底照片,这对一个处女座的PPTer是无法忍受的,那么我们如何用人工智能的方式来解决这个问题呢?


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


官方网址是:

https://github.com/nadermx/backgroundremover


官方介绍是:



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


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


第一步:安装库


pip install backgroundremover


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


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


import osos.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文件就算离线安装好了预训练模型。


样例效果:



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



效果明显更酷炫了。




    
往期精彩回顾





Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/126532