项目背景
孩子出生医生,很多人都有给家里分享照片的需求。之前因为顾虑隐私和数据安全问题,以及商业APP能否持续维护、广告太多等诸多问题,自己用小程序开发了一个宝宝相册,但是不能做到像亲宝宝之类的可以自动识别相册里的宝宝照片并上传,我的小程序需要手动上传照片,而且由于前端开发水平有限,还有一些兼容性问题和上传的bug问题没解决,所以想做一个自动上传的应用。
之前用开源的face recognition模型做了一版,但是这个模型对小孩的识别率很低,测试了一下基本不可用,所以打算在这个模型基础上再训练一个专门针对我家娃的人脸识别模型。
项目计划
业务流程如下图
红色部分是计划训练的人脸识别模型,分别部署在PC和服务器上,用来传本地照片和微信群里其他人发的照片。这里产品经理可能会问了,为啥不直接在手机上识别照片然后自动上传,就不用同步到PC上再上传了。原因是技术能力有限,不会安卓和iOS开发。
人脸识别模型设计如下:
- 模型输入是一张照片,输出是一个照片的分类标签
- 数据集:之前已经整理和上传的孩子的照片
- 模型设计:
- 基于前面提到的开源face recognition模型,找出照片中的人脸部分;
- 链接一个softmax层进行分类(老大、老二、其他),因为只是给自己做的,所以不需要考虑通用的人脸识别优化,用softmax应该足够了。
黑色部分是已经开发好的,如下图