项目计划之--照片自动分类整理归档

小龙 2022年08月11日 1,442次浏览

项目背景

手机拍摄的照片,平常懒得整理,一股脑扔到一个文件夹里,时间长了,要整理,就需要一张一张挨个看,然后分到不同的文件夹。经过一次对电脑的彻底整理,发现这个过程非常耗手指,整理一次,手指疼一次。

项目计划

业务流程如下图

照片归档-1660228885980

其中红色部分是需要开发的,前面照片同步的部分用商业应用就可以。照片分类模型设计如下:

  1. 模型输入是一张照片,输出是一个照片的分类标签
  2. 数据集:之前已经整理的历年照片
  3. 模型设计:
    1. 利用迁移学习,下载其他训练好的模型如Resnet VCG等,然后去掉最后一层,保留前面的特征;
    2. 将上述模型的特征与照片的地理位置信息concat(因为分类有些是根据地理位置分的);
    3. 最后通过一个softmax层进行分类

image-1660229000975

image-1660229072809