项目背景
手机拍摄的照片,平常懒得整理,一股脑扔到一个文件夹里,时间长了,要整理,就需要一张一张挨个看,然后分到不同的文件夹。经过一次对电脑的彻底整理,发现这个过程非常耗手指,整理一次,手指疼一次。
项目计划
业务流程如下图
其中红色部分是需要开发的,前面照片同步的部分用商业应用就可以。照片分类模型设计如下:
- 模型输入是一张照片,输出是一个照片的分类标签
- 数据集:之前已经整理的历年照片
- 模型设计:
- 利用迁移学习,下载其他训练好的模型如Resnet VCG等,然后去掉最后一层,保留前面的特征;
- 将上述模型的特征与照片的地理位置信息concat(因为分类有些是根据地理位置分的);
- 最后通过一个softmax层进行分类