Fashion AI Attributes Recognition of Apparel

天池大数据竞赛——FashionAI全球挑战赛—服饰属性标签识别
每个任务单独训练 + 多任务联合训练融合
比赛对模型数目有限制，本仓库仅限学习
一些经验放在博客，欢迎大家交流学习

Getting Started

代码均在Python3.6下运行，Python2暂未优化测试

环境

Ubuntu16.04
Keras==2.2
tensorflow==1.11
opencv-python==3.4

Python Packages可以通过如下方式快速安装

git clone https://github.com/JosephPai/FashionAI-Attributes.git

cd FashionAI-Attributes-master

pip3 install -r requirement.txt

使用

datasets ---------------------- 存放Annotations标记文件
result ------------------------ 保存模型预测结果
weights ----------------------- 保存模型训练所得权重
config.py --------------------- 配置文件，包括数据集目录等
dataset.py -------------------- 数据预处理
single_task_predict.py -------- 单任务训练脚本
single_task_train.py ---------- 单任务测试脚本
multitask_predict.py ---------- 单任务训练脚本
multitask_train.py ------------ 单任务测试脚本

思路分享

阿里天池服装标签识别比赛新人赛练习经验心得

TODO

multitask train and predict
imgaug库进行data augmentation