话不多说,先上项目链接:
- Github: https://github.com/xiaolin199912/HivisionIDPhotos
- SwanHub在线Demo:https://swanhub.co/ZeYiLin/HivisionIDPhotos/demo

🤩 项目更新
- 2023.6.19: 更新排版照
- 2023.6.13: 更新中心渐变色
- 2023.6.11: 更新上下渐变色
- 2023.6.8: 更新自定义尺寸
- 2023.6.4: 更新自定义底色、人脸检测Bug通知
- 2023.5.10: 更新不改尺寸只换底
这套算法是我当年打互联网+创新创业大赛的时候,开发的一个证件照制作小程序中使用的代码,目前在国内已经被接近100万人使用过了~
把它开源出来主要是因为AIGC浪潮的到来,证件照制作这个非常偏图像代码工程落地的场景就没有那么“性感”;了(好吧应该说有点Old);不过其实也有像Midjourney这样直接生成式解决的办法,比如用户输入一张自拍照,就可以实现一个各种风格的证件照,这样能解决手机自拍做证件照最难处理的面部补光问题;还有像一键换正装,一键换发型等等(当年有做线上海马体的想法来着)。
里面用到的抠图模型是MODNet+网上的开源证件照数据微调的,转成了ONNX来加速推理;人脸检测模型用的是MTCNN,也转成了ONNX,整套代码在CPU上运行的时长还可以接受(不过也有很大的优化空间)。
项目会继续迭代,后续完善一下文章里的算法原理部分,如果友友们觉得有帮助请点个✨star呀~