滑稽，使用paddle轻松搞定抠图，妈妈再也不用担心我不会抠图了

写在前面： 我是「虐猫人薛定谔i」，一个不满足于现状，有梦想，有追求的00后
$\quad$
本博客主要记录和分享自己毕生所学的知识，欢迎关注，第一时间获取更新。
$\quad$
不忘初心，方得始终。
$\quad$

❤❤❤❤❤❤❤❤❤❤

环境搭建

需要安装paddle和paddlehub，如果不会的小伙伴，可以参考paddle的官网。
传送门

快速安装教程

效果展示

代码实现

import os
import paddlehub as hub

# 加载模型
humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')

# 图片路径
path = "D:/Code/Python/res/"

# 判断路径是否存在
if os.path.exists(path):
    files = os.listdir(path)
    imgs = []
    for item in files:
        imgs.append(path + item)
    # 批量抠图
    results = humanseg.segmentation(data={'image': imgs},
                                    visualization=True,
                                    output_dir='D:/Code/Python/result/')
    print("---- 程序结束 ----")
else:
    print("输入的路径不存在！！！")
    print("---- 程序结束 ----")

总结分析

这里说一下我遇到的问题，网上的教程大部分都说，运行完成后，我们可以在项目下看到humanseg_output目录，但是我运行后，代码没有报错，却没有这个目录。其实，这是因为paddle升级了，网上的那是以前的版本，新版本需要自己指定一个目录来存放抠图结果，而且，还要将visualization设置为True。
最后，我还要感谢ZackSock大佬。

蒟蒻写博客不易，加之本人水平有限，写作仓促，错误和不足之处在所难免，谨请读者和各位大佬们批评指正。
如需转载，请署名作者并附上原文链接，蒟蒻非常感激
名称：虐猫人薛定谔i
博客地址：https://blog.csdn.net/Deep___Learning