比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'])

效果展示:

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐