点击上方蓝字关注我们
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.txt
pip 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 8
pip 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