服装标签项目总结与心得
先放项目代码仓库:https://github.com/JosephPai/FashionAI-Attributes
强烈欢迎Star/Follow
关于代码或者数据集如果有任何问题,欢迎大家本博客或者github issues留言一起交流学习
任务描述
从属性标签知识体系中拿出了8种重要的属性维度对服装进行标签识别。
这些属性维度是:颈线设计、领子设计、脖颈设计、翻领设计、袖长、衣长、裙长、裤长。
具体示意,见下图:
针对八个不同的属性维度,分别训练了各自的分类器。
经验总结
从零开始接触这个任务到现在,一些不成熟探索的经验
1.Pre-trained model
最开始接触这个任务的时候,用了一个ResNet101网络直接训练,一方面训练极慢,另一方面效果很差。
虽说训练集有两万多张图片好像看起来不是很少,但是对于CNN提取基础的features来说还不够,
更不用说和ImageNet weights相比。