第一步先安装 paddlehub
pip install --ignore-installed paddlehub -i https://pypi.douban.com/simple/ --user
第二步安装paddlepaddle
pip install --ignore-installed paddlepaddle -i https://pypi.douban.com/simple/ --user
使用官方的测试代码
import paddlehub as hub
# 加载模型,本例为服务器端模型pyramidbox_lite_server_mask
# 移动端模型参数可以换成pyramidbox_lite_mobile_mask
module = hub.Module(name="pyramidbox_lite_server_mask")
# 设置输入数据
input_dict = {"image": ["test_mask_detection.jpg"]}
for data in module.face_detection(data=input_dict):
print(data)
#演示视频https://www.bilibili.com/video/BV1J7411g79B/
运行效果