【人工智能】比PS还好用！paddlehub批量抠图大法好！python

比PS还好用！paddlehub批量抠图大法好！不得不说，这个模型还是非常智能的，可以识别到主体是什么，然后进行智能化抠图处理。

1. 模块下载

python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple
           

pip install paddlehub
           

2. 代码传送门

import os
import sys
import paddlehub as hub

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

# 指定抠图图片目录
path = './image/'
files = []
dirs = os.listdir(path)
for diretion in dirs:
    files.append(path + diretion)

# 抠图
results = humanseg.segmentation(data = {"image":files})

for result in results:
    print(result['origin'])
    print(result['processed'])
           

效果展示：