Spaces:
Running
Running
🤩项目更新
- 在线体验:
- 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: 更新不改尺寸只换底
Overview
🚀谢谢你对我们的工作感兴趣。您可能还想查看我们在图像领域的其他成果，欢迎来信:[email protected].
HivisionIDPhoto旨在开发一种实用的证件照智能制作算法。
它利用一套完善的模型工作流程，实现对多种用户拍照场景的识别、抠图与证件照生成。
HivisionIDPhoto可以做到:
- 轻量级抠图
- 根据不同尺寸规格生成不同的标准证件照、六寸排版照
- 美颜（waiting）
- 智能换正装（waiting）
如果HivisionIDPhoto对你有帮助，请star这个repo或推荐给你的朋友，解决证件照应急制作问题！
🔧环境安装与依赖
- Python >= 3.7（项目主要测试在python 3.10）
- onnxruntime
- OpenCV
- Option: Linux, Windows, MacOS
1. 克隆项目
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd HivisionIDPhotos
2. 安装依赖环境
pip install -r requirements.txt
3. 下载权重文件
在我们的Release下载权重文件hivision_modnet.onnx，存到根目录下。
运行Gradio Demo
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部署
在确保将模型权重文件hivision_modnet.onnx放到根目录下后，在根目录执行：
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
📧联系我们
如果您有任何问题，请发邮件至 [email protected]