告别PS抠图烦恼,开源项目HivisionIDPhotos+PyQt5打造私人证件照工坊
小伙伴们,你是否还在为证件照背景色不对、尺寸不合规而反复跑照相馆?最近,我在GitHub上发现了一个超能打的开源项目——HivisionIDPhotos,它不仅能高精度人像抠图,还能自动生成标准证件照和多尺寸排版照。从命令行一步步推理,到用Gradio搭出Web版,再到最终封装成双击即用的PyQt5桌面应用,全程只用了几个小时。现在,只需上传一张普通照片,选择想要的尺寸(比如一寸、两寸)和背景色(蓝、白、红),程序就会自动完成抠图、换底、排版,还能实时预览并保存结果。支持GPU加速,推理速度飞快,完全本地运行,无需上传云端,隐私又省钱。如果你也受够了手动PS的繁琐,不妨跟着我的折腾笔记,一起打造属于自己的“证件照工作站”。
AI辅助部署
这里我们使用opencode,小伙伴们可以使用claudecod 或者codex、codebuddy、qwen等等
请本地部署开源项目:https://github.com/dot-line/HivisionIDPhotos 包括推理模型的下载,可进行GPU加速推理首先获取代码
给出待办事项
安装依赖
自动下载推理模型
使用方法
命令行推理: cd HivisionIDPhotos 制作证件照 python inference.py -i demo/images/test0.jpg -o ./idphoto.png --height 413 --width 295 人像抠图 python inference.py -t human_matting -i demo/images/test0.jpg -o ./matting.png --matting_model hivision_modnet 启动API服务: cd HivisionIDPhotos python deploy_api.py 访问: http://localhost:8080 启动Gradio Demo: cd HivisionIDPhotos python app.py 访问: http://localhost:7860我们来测试下 demo 图片 test0.jpg
python inference.py -i demo/images/test0.jpg -o ./idphoto.png --height 413 --width 295生成的idphoto.png效果
抠图
python inference.py -t human_matting -i demo/images/test1.jpg -o ./mattingtest1.png --matting_model hivision_modnet添加背景色:
python inference.py -t add_background -i ./idphoto.png -o ./idphoto_ab.jpg -c 4f83ce -k 30 -r 1生成排版照:
python inference.py -t generate_layout_photos -i ./idphoto_ab.jpg -o ./idphoto_layout.jpg --height 413 --width 295 -k 200主界面
AI编程
制作证件照: cd HivisionIDPhotos python inference.py -i demo/images/test0.jpg -o ./idphoto.png --height 413 --width 295 人像抠图: python inference.py -t human_matting -i demo/images/test0.jpg -o ./idphoto_matting.png --matting_model hivision_modnet 添加背景色: python inference.py -t add_background -i ./test1.png -o ./test1_ab.jpg -c 4f83ce -k 30 -r 1 生成排版照: python inference.py -t generate_layout_photos -i ./test1_ab.jpg -o ./test1_ab_layout.jpg --height 413 --width 295 -k 200 根据以上命令,利用 Gradio开发开发 web 应用 用来部署到魔塔社区空间中运行页面
我们发现了一个问题,需要多步骤进行操作,简化下,用户上传照片后自动抠图添加背景色,生成排版照
优化当前功能,用户上传图片,选择相应的证件照尺寸,背景颜色。自动进行抠图添加背景色,生成排版照和证件照。网页版的可能不太方便大家使用,这边改成桌面版,直接双击进行使用
根据当前项目,使用python+pyqt5开发一个证件照制作的界面应用,调用大模型 用户选择路径加载。功能:用户上传一张照片自动裁剪 选择相应的证件照尺寸,背景颜色。自动进行抠图添加背景色,生成排版照和证件照,预览后供用户下载。界面上用户可配置相应的参数。完成桌面应用的开发
感兴趣的小伙们,可评论区留言!