项目优点

  • HivisionIDPhotos是一款开源免费的图片处理工具,可以利用AI模型对照片进行轻量级智能抠图、通过调整尺寸生成不同的标准证件照、替换背景、美颜、智能换正装(后期支持)等操作。

  • 通过它,自己在家也能轻松搞定证件照,而且在本地部署,无需再去网上付费制作证件照,且不需要担心个人信息泄露。

官方在线体验地址

项目介绍

HivisionIDPhoto 旨在开发一种实用、系统性的证件照智能制作算法。

它利用一套完善的AI模型工作流程,实现对多种用户拍照场景的识别、抠图与证件照生成。

HivisionIDPhoto 可以做到:

  1. 轻量级抠图(纯离线,仅需 CPU 即可快速推理)

  2. 根据不同尺寸规格生成不同的标准证件照、六寸排版照

  3. 支持 纯离线 或 端云 推理

  4. 美颜

  5. 智能换正装(waiting)

🤩 最近更新(20241010记录)

  • 在线体验: SwanHub DemoSpaces

  • 2024.09.25: 增加五寸相纸JPEG下载选项|默认照片下载支持300DPI

  • 2024.09.24: API接口增加base64图像传入选项 | Gradio Demo增加排版照裁剪线功能

  • 2024.09.22: Gradio Demo增加野兽模式,可设置内存加载策略 | API接口增加dpi、face_alignment参数

  • 2024.09.18: Gradio Demo增加分享模版照功能、增加美式证件照背景选项

  • 2024.09.17: Gradio Demo增加自定义底色-HEX输入功能 | (社区贡献)C++版本 - HivisionIDPhotos-cpp 贡献 by zjkhahah

  • 2024.09.16: Gradio Demo增加人脸旋转对齐功能,自定义尺寸输入支持毫米单位

  • 2024.09.14: Gradio Demo增加自定义DPI功能,增加日语和韩语支持,增加调整亮度、对比度、锐度功能

  • 2024.09.12: Gradio Demo增加美白功能 | API接口增加加水印设置照片KB值大小证件照裁切

安装教程

Docker Compose方式安装

  1. 演示以飞牛NAS系统为例(其他系统请自行安装好最新版本Docker、docker-compose)

  2. 打开系统的SSH功能,使用终端软件进行连接,并切换到root状态下。
    image-wuqe.png

  3. 在NAS系统的文件管理中新建一个文件夹,右键选择文件详细信息(属性),点击复制原始路径
    (这一步主要是为了获取系统文件夹路径,照顾小白,如果你熟悉的话可以不创建,直接在终端软件内进入这个根目录路径即可。)

  4. 在终端内执行下面的命令,进入文件夹内。

    # 格式:cd 复制的文件夹路径(换成你自己的,别照抄路径。)
    
    cd /vol1/1000/Photo
  5. 克隆项目到本地

    git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
  6. 查看克隆到本地的文件

    ll
  7. 进入项目文件夹

    cd HivisionIDPhotos
  8. 如果你非常熟悉Docker工具,可在查看完项目说明后自行修改docker-compose.yml模板文件。(小白略过这一句)

  9. 编辑docker-compose.yml文件(可选)

    vi docker-compose.yml
  10. 按字母i键进入编辑模式,修改下面的配置文件,按Esc键退出编辑模式,输入:wq退出并保存。(可选)

    version: '3.8'
    
    services:
      hivision_idphotos:
        build:
          context: .
          dockerfile: Dockerfile
        image: linzeyi/hivision_idphotos
        command: python3 -u app.py --host 0.0.0.0 --port 7860
        ports:
          - '7860:7860'    # 第一个7860是容器外部访问端口,可自行修改。
    
      hivision_idphotos_api:
        build:
          context: .
          dockerfile: Dockerfile
        image: linzeyi/hivision_idphotos
        command: python3 deploy_api.py
        ports:
          - '8080:8080'
    
  11. 执行命令,启动docker-compose文件,拉取镜像并创建容器。

    docker compose up -d
    
    或者
    
    docker-compose up -d
  12. 查看正在运行的项目容器

    docker compose ps
    
    或者
    
    docker-compose ps

访问HivisionIDPhotos

  1. 打开浏览器,以NAS的IP+设置的端口进行访问。
    以本机为例:http://192.168.2.225:7860

  2. 使用方式也很简单,直接将你拍好的自拍照放进去。

  3. 根据自己的需要选择好对应的参数,然后点击开始制作。
    在网页底部还有预设的图片,可以自己先点击使用,看看效果。
    (人脸检测模型:face+(联网Online API)这个选项需参考官方文档单独配置才可使用。)

  4. 然后等待系统自动生成即可,点击图片旁边的图标可全屏查看也可保存到本地。

  5. 下面还有社交媒体模板照和抠图照可以用。

  6. 整体用下来效果是不错的,不过图片中过于曝光的白色区域会被系统自动抠除,会显示缺一块。
    同一个照片文件中只允许出现一个人脸,出现两个就会报错,提示你重新上传图片。
    选择不同的模型会消耗不同的性能,实测教程中的图片直接默认配置进行制作,消耗50%CPU3秒(i5-4570),内存消耗约1.5GB。
    如果换了其他的模型/配置,CPU约占50%,内存是直接拉满了,不过在制作完成之后就会恢复正常占用。

项目地址

功德+1(狗子).gif

👇👇👇