tianshuichen/HCP-MLR-PL

Multi-label Image Recognition with Partial Labels

Implementation of papers:

• Structured Semantic Transfer for Multi-Label Recognition with Partial Labels
  36th Association for the Advance of Artificial Intelligence (AAAI), 2022.
  Tianshui Chen, Tao Pu, Hefeng Wu, Yuan Xie, Liang Lin.
• Semantic-Aware Representation Blending for Multi-Label Image Recognition with Partial Labels
  36th Association for the Advance of Artificial Intelligence (AAAI), 2022.
  Tao Pu, Tianshui Chen, Hefeng Wu, Liang Lin.

Preliminary

1. Donwload data.zip, and unzip it.
2. Modify the lines 16-19 in config.py.
3. Create servel folders (i.e., "exp/log", "exp/code", "exp/checkpoint", "exp/summary") to record experiment details.

Usage

1. Run SST

   cd HCP-MLR-PL
   vim scripts/SST.sh
   ./scripts/SST.sh

2. Run SARB

   cd HCP-MLR-PL
   vim scripts/SARB.sh
   ./scripts/SARB.sh

3. Run HST

   cd HCP-MLR-PL
   vim scripts/HST.sh
   ./scripts/HST.sh

Citation

@article{chen2021structured,
  title={Structured Semantic Transfer for Multi-Label Recognition with Partial Labels},
  author={Chen, Tianshui and Pu, Tao and Wu, Hefeng and Xie, Yuan and Lin, Liang},
  journal={arXiv preprint arXiv:2112.10941},
  year={2021}
}

@article{pu2022semantic,
  title={Semantic-Aware Representation Blending for Multi-Label Image Recognition with Partial Labels},
  author={Pu, Tao and Chen, Tianshui and Wu, Hefeng and Lin, Liang},
  journal={arXiv preprint arXiv:2203.02172},
  year={2022}
}

Contributors

For any questions, feel free to open an issue or contact us:

• tianshuichen@gmail.com
• putao537@gmail.com