This respository includes a PyTorch implementation of Task-aware Attention Model for Clothing Attribute Prediction.
python 3.6
PyTorch
Plesae download woman clothing dataset accoding to baidu yun link Fashion dataset
Runing TAN based on AlexNet backbone.
bash train_val_alexnet.sh
Runing TAN based on AlexNet backbone.
bash train_val_vgg16.sh
If this code is helpful for your research, please cite the following paper:
@article{TAN2020,
title={Task-aware Attention Model for Clothing Attribute Prediction},
author={Sanyi Zhang, Zhanjie Song, Xiaochun Cao, Hua Zhang, Jie Zhou},
journal={IEEE Transactions on Circuits and Systems for Video Technology (TCSVT)},
volume={30},
number={4},
pages={1051--1064},
year={2020},
publisher={IEEE}
}