AI个人学习
和实操指南

HivisionIDPhotos:开源智能AI证件照制作工具

综合介绍

HivisionIDPhotos 是一款开源的轻量级 AI 证件照制作工具,能够智能识别用户拍照场景并进行抠图,生成符合多种规格的标准证件照。该工具支持自定义背景色和尺寸,未来还将推出美颜和智能换正装功能。用户可以通过 Docker 部署,轻松搭建本地服务,适用于个人和企业的证件照制作需求。

 

HivisionIDPhotos:智能AI证件照制作工具-1

 

 

功能列表

  • 轻量级抠图:利用高效的 AI 算法,精确抠取人像,生成透明背景图像。
  • 多尺寸证件照生成:支持护照照、签证照等多种规格的证件照生成。
  • 六寸排版照生成:将多张证件照排版成标准的六寸照片,方便打印和批量制作。
  • 智能换背景:支持为证件照增加或更换背景颜色。
  • API 服务:基于 FastAPI 的 API 服务,方便开发者集成证件照功能。
  • Docker 支持:快速搭建和运行服务,简化安装过程,提高系统兼容性。
  • 实时 Web 交互:通过 Gradio Demo 提供的本地 Web 页面,在线操作证件照生成过程。

 

 

使用帮助

安装与环境配置

  1. 克隆项目:
    git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
    cd HivisionIDPhotos
    
  2. 安装依赖环境:
    pip install -r requirements.txt
    
  3. 下载模型文件: 将预训练模型文件 hivision_modnet.onnx 下载并存放到项目根目录下。

 

运行本地服务

  1. 启动 Gradio Demo:
    python app.py
    

    运行程序后,将生成一个本地网页,用户可以在页面中完成证件照的生成和调整。

  2. 部署 API 服务:
    python deploy_api.py
    

    通过 FastAPI 部署 API 服务,用户可以通过 API 实现证件照生成和背景修改等操作。

 

使用 API 服务

  1. 发送证件照制作请求:
    import requests
    
    url = "http://127.0.0.1:8080"
    files = {'file': open('test.jpg', 'rb')}
    response = requests.post(f"{url}/generate_idphoto", files=files)
    with open('idphoto.png', 'wb') as f:
        f.write(response.content)
    
  2. 增加底色:
    import requests
    
    url = "http://127.0.0.1:8080"
    files = {'file': open('idphoto.png', 'rb')}
    data = {'color': '(0,0,0)'}
    response = requests.post(f"{url}/add_background", files=files, data=data)
    with open('idphoto_ab.jpg', 'wb') as f:
        f.write(response.content)
    
  3. 生成六寸排版照:
    import requests
    
    url = "http://127.0.0.1:8080"
    files = {'file': open('idphoto_ab.jpg', 'rb')}
    response = requests.post(f"{url}/generate_layout_photos", files=files)
    with open('idphoto_layout.jpg', 'wb') as f:
        f.write(response.content)
    

 

Docker 部署

  1. 构建 Docker 镜像:
    docker build -t hivisionidphotos .
    
  2. 运行 Docker 容器:
    docker run -p 8080:8080 hivisionidphotos
    

 

 

在线体验

首席AI分享圈此处内容已经被作者隐藏,请输入验证码查看内容
验证码:
请关注本站微信公众号,回复“验证码”,获取验证码。在微信里搜索“首席AI分享圈”或者“Looks-AI”或者微信扫描右侧二维码都可以关注本站微信公众号。

AI轻松学

普通人的AI入门指南

帮助你以低成本、零基础学会如何利用AI工具。AI就像办公软件一样,是每个人的必备技能。 掌握AI会让你在求职中占据优势,在未来的工作和学习中事半功倍。

查看详情>
未经允许不得转载:首席AI分享圈 » HivisionIDPhotos:开源智能AI证件照制作工具
分享到

首席AI分享圈

首席AI分享圈专注于人工智能学习,提供全面的AI学习内容、AI工具和实操指导。我们的目标是通过高质量的内容和实践经验分享,帮助用户掌握AI技术,一起挖掘AI的无限潜能。无论您是AI初学者还是资深专家,这里都是您获取知识、提升技能、实现创新的理想之地。

联系我们