『HivisionIDPhotos』= 极速抠图 + 证件照生成
- 2024-09-25
- 阅读:32
AI证件照制作工具—「HivisionIDPhotos」
-
一键抠图:利用 AI 技术,可以自动从照片中抠出人像,去除背景。
-
多种尺寸和背景:支持多种证件照尺寸和背景颜色的选择,以满足不同证件的要求。
-
智能参数调整:用户可以根据需要调整照片的参数,比如面部比例、头距顶的距离等。
-
支持多种输入方式:除了上传已有的照片,还支持直接使用摄像头拍摄或从剪切板粘贴图片。
-
高速处理:在配备有高性能 CPU 和 GPU 的设备上,可以快速生成证件照。
-
宠物证件照:除了人像证件照, 也可以用来为宠物制作证件照。
GitHub地址:https://github.com/Zeyi-Lin/HivisionIDPhotos
打开Docker仓库,因为dockerhub现在是可以直接使用了,所以这里不需要再设置任何代理或者加速。直接搜索镜像linzeyi/hivision_idphotos
并下载,标签选择latest就可以了。
镜像有一个多GB,因为涉及到了离线人脸检测模型MTCNN和RetinaFace,耐心等待即可。
HivisionIDPhoto 可以做到:
- 轻量级抠图(纯离线,仅需 CPU 即可快速推理)
- 根据不同尺寸规格生成不同的标准证件照、六寸排版照
- 支持 纯离线 或 端云 推理
- 美颜
- 智能换正装(waiting)
以下方式三选一
方式一:拉取最新镜像:
docker pull linzeyi/hivision_idphotos
方式二:Dockrfile 直接构建镜像:
在确保将至少一个抠图模型权重文件放到hivision/creator/weights
下后,在项目根目录执行:
docker build -t linzeyi/hivision_idphotos .
方式三:Docker compose 构建:
在确保将至少一个抠图模型权重文件放到hivision/creator/weights
下后,在项目根目录下执行:
docker compose build
启动 Gradio Demo 服务
运行下面的命令,在你的本地访问 http://127.0.0.1:7860 即可使用。
docker run -d -p 7860:7860 linzeyi/hivision_idphotos
启动 API 后端服务
docker run -d -p 8080:8080 linzeyi/hivision_idphotos python3 deploy_api.py
两个服务同时启动
docker compose up -d
本项目提供了一些额外的配置项,使用环境变量进行设置:
环境变量 | 类型 | 描述 | 示例 |
---|---|---|---|
FACE_PLUS_API_KEY | 可选 | 这是你在 Face++ 控制台申请的 API 密钥 | 7-fZStDJ···· |
FACE_PLUS_API_SECRET | 可选 | Face++ API密钥对应的Secret | VTee824E···· |
RUN_MODE | 可选 | 运行模式,可选值为beast (野兽模式)。野兽模式下人脸检测和抠图模型将不释放内存,从而获得更快的二次推理速度。建议内存16GB以上尝试。 |
beast |
DEFAULT_LANG | 可选 | Gradio Demo启动时的默认语言 | en |
docker使用环境变量示例:
docker run -d -p 7860:7860 \
-e FACE_PLUS_API_KEY=7-fZStDJ···· \
-e FACE_PLUS_API_SECRET=VTee824E···· \
-e RUN_MODE=beast \
-e DEFAULT_LANG=en \
linzeyi/hivision_idphotos
部分评论