MaochengHu 576cda45b8 first commit | 2 jaren geleden | |
---|---|---|
.. | ||
blazeface | 2 jaren geleden | |
demo_images | 2 jaren geleden | |
element_source | 2 jaren geleden | |
solov2 | 2 jaren geleden | |
solov2_blazeface | 2 jaren geleden | |
README.md | 2 jaren geleden | |
test_main.py | 2 jaren geleden | |
test_server.py | 2 jaren geleden |
通过SOLOv2实例分割模型分割人像,并通过BlazeFace关键点模型检测人脸关键点,然后根据两个模型输出结果更换圣诞风格背景并为人脸加上圣诞老人胡子、圣诞眼镜及圣诞帽等特效。本项目通过PaddleHub可直接发布Server服务,供本地调试与前端直接调用接口。您可通过以下二维码中微信小程序直接体验:
paddlepaddle >= 2.0.0rc0
paddlehub >= 2.0.0b1
首先要获取模型,可在模型配置文件里配置solov2
与blazeface_keypoint
,训练模型,并导出模型。也可直接下载我们准备好模型:
blazeface_keypoint模型和
solov2模型。
注意: 下载的模型需要解压后使用。
然后将两个模型文件夹中的文件(infer_cfg.yml
、__model__
和__params__
)分别拷贝至blazeface/blazeface_keypoint/
和 solov2/solov2_r101_vd_fpn_3x/
文件夹内。
hub install solov2
hub install blazeface
$ hub install solov2_blazeface
python test_main.py
运行成功后,预测结果会保存到chrismas_final.png
。
export CUDA_VISIBLE_DEVICES=0
hub serving start -m solov2_blazeface -p 8880
python test_server.py
运行成功后,预测结果会保存到chrismas_final.png
。