HivisionIDPhoto

相关项目：

SwanLab：训练人像抠图模型全程用它来分析和监控，以及和实验室同学协作交流，大幅提升了训练效率。

🤩项目更新

🚀谢谢你对我们的工作感兴趣。您可能还想查看我们在图像领域的其他成果，欢迎来信:zeyi.lin@swanhub.co.

HivisionIDPhoto旨在开发一种实用的证件照智能制作算法。

它利用一套完善的模型工作流程，实现对多种用户拍照场景的识别、抠图与证件照生成。

HivisionIDPhoto可以做到:

如果HivisionIDPhoto对你有帮助，请star这个repo或推荐给你的朋友，解决证件照应急制作问题！

1. 克隆项目

git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd  HivisionIDPhotos

2. 安装依赖环境

pip install -r requirements.txt

3. 下载权重文件

在我们的Release下载权重文件hivision_modnet.onnx，存到根目录下。

python app.py

运行程序将生成一个本地Web页面，在页面中可完成证件照的操作与交互。

python deploy_api.py

请求API服务（Python）

用Python给服务发送请求：

证件照制作（输入1张照片，获得1张标准证件照和1张高清证件照的4通道透明png）：

python requests_api.py -u http://127.0.0.1:8080 -i test.jpg -o ./idphoto.png -s '(413,295)'

增加底色（输入1张4通道透明png，获得1张增加了底色的图像）：

python requests_api.py -u http://127.0.0.1:8080 -t add_background -i ./idphoto.png -o ./idhoto_ab.jpg  -c '(0,0,0)'

得到六寸排版照（输入1张3通道照片，获得1张六寸排版照）：

python requests_api.py -u http://127.0.0.1:8080 -t generate_layout_photos -i ./idhoto_ab.jpg -o ./idhoto_layout.jpg  -s '(413,295)'

在确保将模型权重文件hivision_modnet.onnx放到根目录下后，在根目录执行：

docker build -t hivision_idphotos .

等待镜像封装完毕后，运行以下指令，即可开启API服务：

docker run -p 8080:8080 hivision_idphotos

如果您有任何问题，请发邮件至 zeyi.lin@swanhub.co