Spaces:
Running
Running
File size: 3,853 Bytes
d5d20be b306945 d5d20be |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 |
<div align="center">
<h1>HivisionIDPhoto</h1>
[English](README_EN.md) / 中文
[](https://github.com/xiaolin199912/HivisionIDPhotos)
[](https://swanhub.co/ZeYiLin/HivisionIDPhotos/demo)
[](https://zhuanlan.zhihu.com/p/638254028)
<img src="assets/demoImage.png" width=900>
</div>
# 🤩项目更新
- 在线体验: [](https://swanhub.co/ZeYiLin/HivisionIDPhotos/demo)
- 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可以做到:**
1. 轻量级抠图
2. 根据不同尺寸规格生成不同的标准证件照、六寸排版照
3. 美颜（waiting）
4. 智能换正装（waiting）
<div align="center">
<img src="assets/gradio-image.jpeg" width=900>
</div>
---
如果HivisionIDPhoto对你有帮助，请star这个repo或推荐给你的朋友，解决证件照应急制作问题！
<br>
# 🔧环境安装与依赖
- Python >= 3.7（项目主要测试在python 3.10）
- onnxruntime
- OpenCV
- Option: Linux, Windows, MacOS
**1. 克隆项目**
```bash
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd HivisionIDPhotos
```
**2. 安装依赖环境**
```bash
pip install -r requirements.txt
```
**3. 下载权重文件**
在我们的[Release](https://github.com/Zeyi-Lin/HivisionIDPhotos/releases/tag/pretrained-model)下载权重文件`hivision_modnet.onnx`，存到根目录下。
<br>
# 运行Gradio Demo
```bash
python app.py
```
运行程序将生成一个本地Web页面，在页面中可完成证件照的操作与交互。
<br>
# 部署API服务
```
python deploy_api.py
```
**请求API服务（Python）**
用Python给服务发送请求：
证件照制作（输入1张照片，获得1张标准证件照和1张高清证件照的4通道透明png）：
```bash
python requests_api.py -u http://127.0.0.1:8080 -i test.jpg -o ./idphoto.png -s '(413,295)'
```
增加底色（输入1张4通道透明png，获得1张增加了底色的图像）：
```bash
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张六寸排版照）：
```bash
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)'
```
<br>
# 🐳Docker部署
在确保将模型权重文件[hivision_modnet.onnx](https://github.com/Zeyi-Lin/HivisionIDPhotos/releases/tag/pretrained-model)放到根目录下后，在根目录执行：
```bash
docker build -t hivision_idphotos .
```
等待镜像封装完毕后，运行以下指令，即可开启API服务：
```bash
docker run -p 8080:8080 hivision_idphotos
```
<br>
# 引用项目
1. MTCNN: https://github.com/ipazc/mtcnn
2. ModNet: https://github.com/ZHKKKe/MODNet
<br>
# 📧联系我们
如果您有任何问题，请发邮件至 [email protected] |