8K star!开源的小巧工具,帮你实现证件照自由

文摘   科技   2024-09-10 11:06   北京  

点击上方蓝字关注我们

IT 咖啡馆,探索无限可能!

恭喜你发现了这个宝藏,这里你会发现优质的开源项目、IT知识和有趣的内容。

 

如果哈利波特在中国要上学,他得先准备什么?魔法棒、猫头鹰还是黑斗篷?不!他得先照张证件照。

今天我们分享的开源项目,它可以快速帮忙搞定证件照,几秒钟就能一键生成各规格的证件照,它就是:HivisionIDPhotos




HivisionIDPhotos 是什么



HivisionIDPhotos是一个通过AI来实现证件照的工具,它能够帮助你快速生成符合不同规格的标准证件照,并且支持多种功能,如智能抠图、美颜(即将推出)以及智能换正装(即将推出)。而且只需要CPU就可以完成抠图和生成工作,并且支持 纯离线 或 端云 推理。

项目的特点包括了以下:

  • 自动化处理:自动检测人脸并进行裁剪,以符合证件照的标准尺寸要求。

  • 背景替换:自动更换照片背景,确保照片背景符合证件照的背景颜色要求。

  • 图像增强:图像质量增强功能,提高照片的整体质量。

  • 尺寸调整:提供多种标准证件照尺寸选项,可以根据需要选择合适的尺寸。

  • 易于使用:用户界面友好,简单易用轻松上手。

  • 高效性:仅需CPU即可快速推理,减少用户等待时间。

 




安装使用


 

在线demo

如果只是想简单了解项目的功能,或是快速使用一下,那么可以考虑先试用项目提供的在线demo

https://swanhub.co/ZeYiLin/HivisionIDPhotos/demo
 

本地安装

项目支持在Linux、windows、MacOS上安装,安装前需要确认已经安装Python3.7以上版本,建议使用Python3.10。

 

克隆项目

git clone <https://github.com/Zeyi-Lin/HivisionIDPhotos.git>cd  HivisionIDPhotos

 

安装依赖环境

 

建议 conda 创建一个 python3.10 虚拟环境后,执行以下命令

pip install -r requirements.txtpip install -r requirements-app.txt

 

下载权重文件

 

安装完成依赖后,需要下载相关的模型权重文件,可以通过脚本来下载或者直接自己手动下载。

方式一:脚本下载

python scripts/download_model.py --models all

方式二:直接下载

存到项目的hivision/creator/weights目录下:

  • modnet_photographic_portrait_matting.onnx (24.7MB): MODNet官方权重,下载

  • hivision_modnet.onnx (24.7MB): 对纯色换底适配性更好的抠图模型,下载

  • rmbg-1.4.onnx (176.2MB): BRIA AI 开源的抠图模型,下载后重命名为rmbg-1.4.onnx

  • birefnet-v1-lite.onnx(224MB): ZhengPeng7 开源的抠图模型,下载后重命名为birefnet-v1-lite.onnx

 

GPU推理加速

 

如需使用英伟达GPU加速推理,在确保你已经安装CUDA与cuDNN后,根据文档找到对应的onnxruntime-gpu版本安装,如:

# CUDA 12.x, cuDNN 8pip install onnxruntime-gpu==1.18.0

完成后,调用如birefnet-v1-lite模型将会利用GPU加速推理。

启动WEB UI

python app.py





社区扩展




HivisionIDPhotos 的易用性让很多社区网友积极的参与进来,很快就衍生出了一些有趣的扩展项目,比如:

HivisionIDPhotos-windows-GUI:是基于 HivisionIDPhotos 开发的Windows客户端应用, 支持任意人像图片,自动做成不同尺寸证件照,支持不同底色,批量生成。

HivisionIDPhotos-ComfyUI:是ComfyUI证件照处理工作流,在AI生成图片领域ComfyUI非常流行,所以很快就有人制作除了HivisionIDPhotos的工作流节点,可以将其加入到自己的流程中。




总结



总的来说Dokku是一个非常有用的工具,虽然现在的PaaS平台非常强大,比如vercal、firebase等,但如果你不想受制于人或是想完全的掌握自己的数据,那么可以通过Dokku来构建自己的平台,能力也很强大。另外你可以通过它的起来,来考虑构建和实现自己的PaaS来提供给别人使用。

   




项目信息



  • 项目名称:HivisionIDPhotos

  • GitHub 链接:https://github.com/Zeyi-Lin/HivisionIDPhotos

  • Star 数:8K



往期推荐

28K star!开源的自建PaaS平台,拥有自己的Heroku

8K star!开源行为检测引擎,更适合云原生环境

「Github一周热点39期」开源RAG UI、AI证件照、开源免费ERP等5个项目

  




识别二维码关注我们
微信号- it-coffee
B站 - IT-咖啡馆
头条号- IT咖啡馆

IT咖啡馆
开源项目、IT技能和有趣的事情
 最新文章