🤩项目更新
- 在线体验:
- 2023.12.1: 更新API部署(基于fastapi)
- 2023.6.20: 更新预设尺寸菜单
- 2023.6.19: 更新排版照
- 2023.6.13: 更新中心渐变色
- 2023.6.11: 更新上下渐变色
- 2023.6.8: 更新自定义尺寸
- 2023.6.4: 更新自定义底色、人脸检测Bug通知
- 2023.5.10: 更新不改尺寸只换底
概述
🚀感谢您对我们工作的兴趣。您可能还想了解我们在图像领域的其他成果,欢迎来信:zeyi.lin@swanhub.co。
HivisionIDPhoto旨在开发一种实用的证件照智能制作算法。
它利用一套完善的模型工作流程,实现对多种用户拍照场景的识别、抠图与证件照生成。
HivisionIDPhoto可以实现:
- 轻量级抠图
- 根据不同尺寸规格生成不同的标准证件照、六寸排版照
- 美颜(开发中)
- 智能换正装(开发中)
如果HivisionIDPhoto对您有帮助,请为这个仓库点星或推荐给您的朋友,解决证件照应急制作问题!
🔧环境安装与依赖
安装
- 克隆仓库
git lfs install && git clone https://swanhub.co/ZeYiLin/HivisionIDPhotos.git
cd HivisionIDPhotos
- 安装依赖包
pip install -r requirements.txt
⚡️快速推理
模型与代码通过git-lfs下载。
git lfs install
git clone https://swanhub.co/ZeYiLin/HivisionIDPhotos.git
演示
python app.py
运行程序将生成一个本地Web页面,在页面中可完成证件照的操作与交互。
部署API服务
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)'
🐳Docker部署
在根目录下执行:
docker build -t hivision_idphotos .
等待镜像封装完毕后,运行以下指令,即可开启API服务:
docker run -p 8080:8080 hivision_idphotos
引用项目
- MTCNN: https://github.com/ipazc/mtcnn
- ModNet: https://github.com/ZHKKKe/MODNet
📧联系我们
如果您有任何问题,请发邮件至 zeyi.lin@swanhub.co