怎么用Python代码批量抠图
导读:本文共1560.5字符,通常情况下阅读需要5分钟。同时您也可以点击右侧朗读,来听本文内容。按键盘←(左) →(右) 方向键可以翻页。
摘要: 1.准备为了实现这个实验,Python是必不可少的,如果你还没有安装Python,可以关注文末的微信公众号获取下载安装指南然后,我们需要安装baidu的paddlepaddle, 官方网站就有详细的指引:https://www.paddlepaddle.org.cn/install/quick根据你自己的情况选择这些选项,最后一个CUDA版本,由于本实验不需要训... ...
目录
(为您整理了一些要点),点击可以直达。1.准备
为了实现这个实验,Python是必不可少的,如果你还没有安装Python,可以关注文末的微信公众号获取下载安装指南
然后,我们需要安装baidu的paddlepaddle, 官方网站就有详细的指引:
https://www.paddlepaddle.org.cn/install/quick
根据你自己的情况选择这些选项,最后一个CUDA版本,由于本实验不需要训练数据,也不需要太大的计算量,所以直接选择CPU版本即可。选择完毕,下方会出现安装指引,不得不说,Paddlepaddle这些方面做的还是比较贴心的(小声bb:就是名字起的不好)。
要注意,如果你的Python3环境变量里的程序名称是Python,记得将语句改为Python xxx,如下进行安装:
python-mpipinstallpaddlepaddle-ihttps://mirror.baidu.com/pypi/simple
还需要安装paddlehub:
pipinstall-ihttps://mirror.baidu.com/pypi/simplepaddlehub
2.编写代码
整个步骤分为三步:
加载模型
指定待抠图的图片目录
抠图
importosimportsysimportpaddlehubashub#1.加载模型humanseg=hub.Module(name="deeplabv3p_xception65_humanseg")#2.指定待抠图图片目录path='./source/'files=[]dirs=os.listdir(path)fordiretionindirs:files.append(path+diretion)#3.抠图results=humanseg.segmentation(data={"image":files})forresultinresults:print(result['origin'])print(result['processed'])
不多不少一共20行代码。抠图完毕后会在本地文件夹下产生一个叫做humanseg_output的文件夹。这里面存放的是已经抠图成功的图片。
3.结果分析
不得不承认,谷歌的算法就素厉害啊。只要背景好一点,抠出来的细节都和手动抠的细节不相上下,甚至优于人工手段。
不过在背景和人的颜色不相上下的情况下,会产生一些问题,比如下面这个结果:
背后那个大叔完全被忽略掉了(求大叔的内心阴影面积)。尽管如此,这个模型是我迄今为止见过的最强抠图模型,没有之一。
</div> <div class="zixun-tj-product adv-bottom"></div> </div> </div> <div class="prve-next-news">
怎么用Python代码批量抠图的详细内容,希望对您有所帮助,信息来源于网络。