attribute-aware-attention

by iamhankai

[ACM MM 2018] Attribute-Aware Attention Model for Fine-grained Representation Learning

137 Stars 28 Forks Last release: Not found 16 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

Attribute-Aware Attention Model

Code for ACM Multimedia 2018 oral paper: Attribute-Aware Attention Model for Fine-grained Representation Learning

We have presented results of fine-grained classification, person re-id, image retrieval tasks, including CUB-200-2011, Market-1501, CARS196 datasets in the paper. Here is the example of fine-grained classification. For detailed results, refer to the original paper or ArXiv.

Usage

Requires: Keras 1.2.1 ("imagedataformat": "channels_first")

Run in two steps:

  1. Download CUB-200-2011 dataset here and unzip it to
    $CUB
    ; Copy file
    tools/processed_attributes.txt
    to
    $CUB
    .
  • The
    $CUB
    dir should be like this:
  1. Change
    data_dir
    in
    run.sh
    to
    $CUB
    , run the scprit
    sh run.sh
    to obtain the result.
  • Result on CUB dataset

Citation

Please use the following bibtex to cite our work:

@inproceedings{han2018attribute,
  title={Attribute-Aware Attention Model for Fine-grained Representation Learning},
  author={Han, Kai and Guo, Jianyuan and Zhang, Chao and Zhu, Mingjian},
  booktitle={Proceedings of the 26th ACM international conference on Multimedia},
  pages={2040--2048},
  year={2018},
  organization={ACM}
}

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.