xiaolin199912/hivisionidphotos

⚡️HivisionIDPhotos: a lightweight and efficient AI ID photos tools. 一个轻量级的AI证件照制作算法。

HivisionIDPhoto

English / 中文

🤩项目更新

• 在线体验:
• 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）

---

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

🔧环境安装与依赖

• Python >= 3.7 (Recommend to use Anaconda or Miniconda)
• onnxruntime
• OpenCV
• Option: Linux, Windows, MacOS

Installation

1. Clone repo

git lfs install && git clone https://swanhub.co/ZeYiLin/HivisionIDPhotos.git
cd  HivisionIDPhotos

2. Install dependent packages

pip install -r requirements.txt

⚡️快速推理

模型与代码通过git-lfs下载。

git lfs install
git clone https://swanhub.co/ZeYiLin/HivisionIDPhotos.git

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部署

在根目录下执行：

docker build -t hivision_idphotos .

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

docker run -p 8080:8080 hivision_idphotos

引用项目

1. MTCNN: https://github.com/ipazc/mtcnn
2. ModNet: https://github.com/ZHKKKe/MODNet

📧联系我们

如果您有任何问题，请发邮件至 [email protected]

hivisionidphotos's Issues

头发比较高的证件照能不能给它压扁一点？

这个有什么方法吗？

运行时出现问题

Task exception was never retrieved
future: <Task finished coro=<Queue.process_events() done, defined at D:\ProgramData\anaconda3\envs\yologesture\lib\site-packages\gradio\queueing.py:343> exception=1 validation error for PredictBody
event_id
Field required [type=missing, input_value={'data': ['只换底'], '...on_hash': 'd0dr5fhazce'}, input_type=dict]
For further information visit https://errors.pydantic.dev/2.4/v/missing>
Traceback (most recent call last):
File "D:\ProgramData\anaconda3\envs\yologesture\lib\site-packages\gradio\queueing.py", line 347, in process_events
client_awake = await self.gather_event_data(event)
File "D:\ProgramData\anaconda3\envs\yologesture\lib\site-packages\gradio\queueing.py", line 220, in gather_event_data
data, client_awake = await self.get_message(event, timeout=receive_timeout)
File "D:\ProgramData\anaconda3\envs\yologesture\lib\site-packages\gradio\queueing.py", line 456, in get_message
return PredictBody(**data), True
File "D:\ProgramData\anaconda3\envs\yologesture\lib\site-packages\pydantic\main.py", line 164, in init
pydantic_self.pydantic_validator.validate_python(data, self_instance=pydantic_self)
pydantic_core._pydantic_core.ValidationError: 1 validation error for PredictBody
event_id
Field required [type=missing, input_value={'data': ['只换底'], '...on_hash': 'd0dr5fhazce'}, input_type=dict]
For further information visit https://errors.pydantic.dev/2.4/v/missing

pip install -r requirements.txt 出错

pip install -r requirements.txt 出错
Requirement already satisfied: numpy in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 4)) (1.19.5)
ERROR: Could not find a version that satisfies the requirement gradio==3.38.0 (from versions: 0.1.0, 0.1.1, 0.1.2, 0.1.3, 0.1.4, 0.1.5, 0.1.6, 0.1.7, 0.1.8, 0.1.9, 0.2.0, 0.2.1, 0.3.0, 0.3.1, 0.3.2, 0.3.3, 0.3.4, 0.3.5, 0.4.0, 0.4.1, 0.4.2, 0.4.4, 0.5.0, 0.7.0, 0.7.1, 0.7.2, 0.7.3, 0.7.4, 0.7.5, 0.7.6, 0.7.7, 0.7.8, 0.8.0, 0.8.1, 0.9.0, 0.9.1, 0.9.2, 0.9.3, 0.9.4, 0.9.5, 0.9.6, 0.9.7, 0.9.8, 0.9.9.2, 0.9.9.3, 0.9.9.5, 0.9.9.6, 0.9.9.7, 0.9.9.8, 0.9.9.9, 0.9.9.9.2, 1.0.0a1, 1.0.0a3, 1.0.0a4, 1.0.0, 1.0.1, 1.0.2, 1.0.3, 1.0.4, 1.0.5, 1.0.6, 1.0.7, 1.1.0, 1.1.1, 1.1.2, 1.1.3, 1.1.4, 1.1.5, 1.1.6, 1.1.8, 1.1.8.1, 1.1.9, 1.2.2, 1.2.3, 1.3.0, 1.3.1, 1.3.2, 1.4.0, 1.4.2, 1.4.3, 1.4.4, 1.5.0, 1.5.1, 1.5.3, 1.5.4, 1.6.0, 1.6.1, 1.6.2, 1.6.3, 1.6.4, 1.7.0, 1.7.1, 1.7.2, 1.7.3, 1.7.4, 1.7.5, 1.7.6, 1.7.7, 2.0.0, 2.0.1, 2.0.2, 2.0.4, 2.0.5, 2.0.6, 2.0.7, 2.0.8, 2.0.9, 2.0.10, 2.1.0, 2.1.1, 2.1.2, 2.1.4, 2.1.6, 2.1.7, 2.2.0, 2.2.1, 2.2.2, 2.2.3, 2.2.4, 2.2.5, 2.2.6, 2.2.7, 2.2.8, 2.2.9a0, 2.2.9a2, 2.2.10, 2.2.11, 2.2.12, 2.2.13, 2.2.14, 2.2.15, 2.3.0a0, 2.3.0b99, 2.3.0b101, 2.3.0b102, 2.3.0, 2.3.3, 2.3.4, 2.3.5b0, 2.3.5, 2.3.6, 2.3.7b0, 2.3.7b1, 2.3.7b2, 2.3.7, 2.3.8b0, 2.3.9, 2.4.0a0, 2.4.0, 2.4.1, 2.4.2, 2.4.4, 2.4.5, 2.4.6, 2.4.7b0, 2.4.7b2, 2.4.7b3, 2.4.7b4, 2.4.7b5, 2.4.7b6, 2.4.7b7, 2.4.7b8, 2.4.7b9, 2.5.0, 2.5.1, 2.5.2, 2.5.3, 2.5.8a0, 2.6.0, 2.6.1a0, 2.6.1b0, 2.6.1b3, 2.6.1, 2.6.2, 2.6.3, 2.6.4b0, 2.6.4b2, 2.6.4b3, 2.6.4, 2.7.0a101, 2.7.0a102, 2.7.0b70, 2.7.0, 2.7.5, 2.7.5.1, 2.7.5.2b0, 2.7.5.2, 2.8.0a100, 2.8.0b0, 2.8.0b2, 2.8.0b3, 2.8.0b4, 2.8.0b5, 2.8.0b6, 2.8.0b10, 2.8.0b12, 2.8.0b20, 2.8.0b22, 2.8.0, 2.8.1, 2.8.2, 2.8.3, 2.8.4, 2.8.5, 2.8.6, 2.8.7, 2.8.8, 2.8.9, 2.8.10, 2.8.11, 2.8.12, 2.8.13, 2.8.14, 2.9.0b0, 2.9.0b1, 2.9.0b2, 2.9.0b3, 2.9.0b5, 2.9.0b6, 2.9.0b7, 2.9.0b8, 2.9.0b9, 2.9.0b10, 2.9b11, 2.9b12, 2.9b13, 2.9b14, 2.9b15, 2.9b20, 2.9b21, 2.9b22, 2.9b23, 2.9b24, 2.9b25, 2.9b26, 2.9b27, 2.9b28, 2.9b30, 2.9b31, 2.9b32, 2.9b33, 2.9b40, 2.9b48, 2.9b50, 2.9.0, 2.9.0.1, 2.9.1, 2.9.2, 2.9.3, 2.9.4, 3.0b0, 3.0b1, 3.0b2, 3.0b5, 3.0b6, 3.0b8, 3.0b9, 3.0b10, 3.0, 3.0.1, 3.0.2, 3.0.3, 3.0.4, 3.0.5, 3.0.6b1, 3.0.6b2, 3.0.6b3, 3.0.6, 3.0.7, 3.0.8b1, 3.0.8, 3.0.9b10, 3.0.9b11, 3.0.9b20, 3.0.9, 3.0.10, 3.0.11b1, 3.0.11, 3.0.12)
ERROR: No matching distribution found for gradio==3.38.0

只换底色之后，再搞寸照之类，六寸排版好像没有了，要刷新

API是不是还没有达到app的效果啊？

什么时候支持换正装

什么时候支持换正装

用上了非常不错

用上了非常不错，如果有美颜，调色美颜就完美了

人脸数量不等于1如何解决，图中只有一个人脸啊

如题

在docker中部署以后网页404 not found。

您好，我在docker中编译并完成部署以后，点击打开对应的网站，显示404not found，请问是什么原因导致的呢？

换装的话是否替换背景？

运行报错：Load model from ./hivision_modnet.onnx failed:Protobuf parsing failed