『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 可以做到:

  1. 轻量级抠图(纯离线,仅需 CPU 即可快速推理)
  2. 根据不同尺寸规格生成不同的标准证件照、六寸排版照
  3. 支持 纯离线 或 端云 推理
  4. 美颜
  5. 智能换正装(waiting)

Docker 部署

1. 拉取或构建镜像

以下方式三选一

方式一:拉取最新镜像:

docker pull linzeyi/hivision_idphotos
 

方式二:Dockrfile 直接构建镜像:

在确保将至少一个抠图模型权重文件放到hivision/creator/weights下后,在项目根目录执行:

docker build -t linzeyi/hivision_idphotos .
 

方式三:Docker compose 构建:

在确保将至少一个抠图模型权重文件放到hivision/creator/weights下后,在项目根目录下执行:

docker compose build
 

2. 运行服务

启动 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  
获取资源方式一
请登录后查看下载方法

立即登录

部分评论

阿里云优惠购