/Image-Attribute-Recognition

Image attribute recognition based on ALM

Primary LanguagePython

Image Attribute Recognition

Image attribute recognition code implementation based on ALM model [1]

Requirements

pip install -r requirements.txt

Usages

  1. Download the trained model from here.
  2. Put the model saved_model.pt file under ./model or ./app/alm/model
  3. Train 
    python main.py --experiment=rap
  4. Visualize 
    python main.py --experiment=rap --vis
    Start TensorBoard with tensorboard --logdir ./runs

Reference

[1] Tang, C., Sheng, L., Zhang, Z., & Hu, X. (2019). Improving pedestrian attribute recognition with weakly-supervised multi-scale attribute-specific localization. In Proceedings of the IEEE/CVF International Conference on Computer Vision (pp. 4997-5006).